remove hard coded input from category file generation (#4841)

fa786cce · Philip Meier · GitHub · 0817f7f5 · fa786cce
Unverified Commit fa786cce authored Nov 03, 2021 by Philip Meier Committed by GitHub Nov 03, 2021
Show 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
@@ -52,7 +52,7 @@ def parse_args(argv=None):


 if __name__ == "__main__":
-    args = parse_args(["-f", "sbd"])
+    args = parse_args()

    try:
        main(*args.names, force=args.force)