args.path is never not set

pytorch-labs · Sep 3, 2024 · c1f163c · c1f163c
1 parent cefe3ee
commit c1f163c
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/torchfix/__main__.py b/torchfix/__main__.py
@@ -40,7 +40,7 @@ def StderrSilencer(redirect: bool = True):
             libc.close(orig_stderr)
 
 
-def main() -> None:
+def _parse_args() -> argparse.Namespace:
     parser = argparse.ArgumentParser()
 
     parser.add_argument(
@@ -78,11 +78,11 @@ def main() -> None:
         action="store_true",
     )
 
-    args = parser.parse_args()
+    return parser.parse_args()
 
-    if not args.path:
-        parser.print_usage()
-        sys.exit(1)
+
+def main() -> None:
+    args = _parse_args()
 
     files = codemod.gather_files(args.path)