从 Jupyter 笔记本中的 Python 脚本导入时出现无法识别的 arguments 异常

Question

我不确定这是一个错误还是我做错了什么。

我有一个独立的 python 脚本，它使用 argparse 从命令行启动。

我把它弄傻了一点，但这本质上是脚本：

测试.py

import argparse

parser = argparse.ArgumentParser()
parser.add_argument("-i", "--inputfile", type=str, default=None,
                    help="filename containing input data")
parser.add_argument("-o", "--outputfile", type=str, default=None,
                    help="filename to save output data to")
args = parser.parse_args()


def step_test_experiment(inputs_filename=None, outputs_filename=None):
    print(inputs_filename, outputs_filename)

if __name__ == '__main__':

    if args.outputfile is None:
        outputs_filename = "tclab_dyn_data.csv"
    else:
        outputs_filename = args.outputfile
    step_test_experiment(args.inputfile, outputs_filename)

当我在 Jupyter 笔记本中尝试以下操作时：

from test import step_test_experiment

我得到：

usage: ipykernel_launcher.py [-h] [-i INPUTFILE] [-o OUTPUTFILE]
ipykernel_launcher.py: error: unrecognized arguments: -f /Users/billtubbs/Library/Jupyter/runtime/kernel-df168fd5-a7f9-4258-b106-0e4582601e79.json
An exception has occurred, use %tb to see the full traceback.

SystemExit: 2


//anaconda3/envs/torch/lib/python3.7/site-packages/IPython/core/interactiveshell.py:3334: UserWarning: To exit: use 'exit', 'quit', or Ctrl-D.
  warn("To exit: use 'exit', 'quit', or Ctrl-D.", stacklevel=1)

%tb产生：

---------------------------------------------------------------------------
SystemExit                                Traceback (most recent call last)
<ipython-input-1-14c395429157> in <module>
----> 1 from test import step_test_experiment

~/dd-temp-control/test.py in <module>
      6 parser.add_argument("-o", "--outputfile", type=str, default=None,
      7                     help="filename to save output data to")
----> 8 args = parser.parse_args()
      9 
     10 

//anaconda3/envs/torch/lib/python3.7/argparse.py in parse_args(self, args, namespace)
   1750         if argv:
   1751             msg = _('unrecognized arguments: %s')
-> 1752             self.error(msg % ' '.join(argv))
   1753         return args
   1754 

//anaconda3/envs/torch/lib/python3.7/argparse.py in error(self, message)
   2499         self.print_usage(_sys.stderr)
   2500         args = {'prog': self.prog, 'message': message}
-> 2501         self.exit(2, _('%(prog)s: error: %(message)s\n') % args)

//anaconda3/envs/torch/lib/python3.7/argparse.py in exit(self, status, message)
   2486         if message:
   2487             self._print_message(message, _sys.stderr)
-> 2488         _sys.exit(status)
   2489 
   2490     def error(self, message):

SystemExit: 2

看起来 Jupyter kernel 在加载脚本时添加了一个“-f”参数。

任何想法是什么原因造成的？

Answer 1

如果，正如你在评论中所说，

在这种情况下，我实际上不想使用 argparser

解决方案是将整个参数解析块移动到if __name__ == "__main__":块中，结果是：

def step_test_experiment(inputs_filename=None, outputs_filename=None):
    print(inputs_filename, outputs_filename)

if __name__ == '__main__':

    import argparse

    parser = argparse.ArgumentParser()
    parser.add_argument("-i", "--inputfile", type=str, default=None,
                    help="filename containing input data")
    parser.add_argument("-o", "--outputfile", type=str, default=None,
                    help="filename to save output data to")
    args = parser.parse_args()
    if args.outputfile is None:
        outputs_filename = "tclab_dyn_data.csv"
    else:
        outputs_filename = args.outputfile
    step_test_experiment(args.inputfile, outputs_filename)