fix nargs specifier for category file generation (#4663)

341fca91 · Philip Meier · GitHub · 453199bf · 341fca91
Unverified Commit 341fca91 authored Oct 20, 2021 by Philip Meier Committed by GitHub Oct 20, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/prototype/datasets/generate_category_files.py torchvision/prototype/datasets/generate_category_files.py +1 -1

No files found.
--- 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.",
    )