Исключение нераспознанных аргументов при импорте из скрипта Python в записной книжке Jupyter - PullRequest
Новая
       17

Исключение нераспознанных аргументов при импорте из скрипта Python в записной книжке Jupyter

0 голосов
/ 23 октября 2019

Я не уверен, что это ошибка или я что-то не так делаю.

У меня есть автономный скрипт на python, который использует argparse для запуска из командной строки.

Я немного ошарашился, но это по сути сценарий:

test.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 добавляет аргумент '-f', когда онозагружает скрипт.

Есть идеи, что вызывает это?

1 Ответ

1 голос
/ 23 октября 2019

Если, как вы сказали в комментариях,

Я на самом деле не хочу использовать 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)
...