【Google colab】argparseでエラーが出た場合の対処法

環境

Google Colaboratory

現象

以下のコードを Google colab で実行すると、次のようなエラーが出る

Python

import argparse

parser = argparse.ArgumentParser(description="argparse test")
parser.add_argument('-me', '-max_epoch', type=int, default=200)
parser.add_argument('-ms', '-minibatch_size', type=int, default=256)
parser.add_argument('-g', '-gpu', type=int, default=-1)
parser.add_argument('-path', default='./')
args = parser.parse_args()

print(args)

出力

usage: ipykernel_launcher.py [-h] [-me ME] [-ms MS] [-g G] [-path PATH]
ipykernel_launcher.py: error: unrecognized arguments: -f /root/.local/share/jupyter/runtime/kernel-d19b24d6-a0cd-4fbc-9299-7feb199b5ddc.json
An exception has occurred, use %tb to see the full traceback.

SystemExit: 2

対処法

上のコードで args = parser.parse_args() を args = parser.parse_args(args=[]) に変えるとエラーが消える

出力

Namespace(g=-1, me=200, ms=256, path='./')

これで args.xx が使えるようになる

参考