fix nargs specifier for category file generation (#4663)

pmeier · web-flow · commit 341fca9154a3 · 2021-10-20T10:27:37.000+02:00
diff --git a/torchvision/prototype/datasets/generate_category_files.py b/torchvision/prototype/datasets/generate_category_files.py
@@ -38,7 +38,7 @@ def parse_args(argv=None):
 
     parser.add_argument(
         "names",
-        nargs="?",
+        nargs="*",
         type=str,
         help="Names of datasets to generate category files for. If omitted, all datasets will be used.",
     )

Original file line number	Diff line number	Diff line change
`@@ -38,7 +38,7 @@ def parse_args(argv=None):`
`38`	`38`
`39`	`39`	`parser.add_argument(`
`40`	`40`	`"names",`
`41`		`- nargs="?",`
	`41`	`+ nargs="*",`
`42`	`42`	`type=str,`
`43`	`43`	`help="Names of datasets to generate category files for. If omitted, all datasets will be used.",`
`44`	`44`	`)`