multipy

Reviewed By: zsol Differential Revision: D51343327 fbshipit-source-id: 83b7a51a916c6f4949a98d5586e2ca3d7ee93b49
pytorch · Nov 15, 2023 · 055676d · 055676d
1 parent 765dfef
commit 055676d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/multipy/runtime/example/generate_examples.py b/multipy/runtime/example/generate_examples.py
@@ -88,7 +88,8 @@ def save(
 parser.add_argument("--install_dir", help="Root directory for all output files")
 
 
-if __name__ == "__main__":
+def main() -> None:
+    global p
     args = parser.parse_args()
     if args.install_dir is None:
         p = Path(__file__).parent / "generated"
@@ -150,3 +151,7 @@ def save(
         e.mock("iopath.**")
         e.intern("**")
         e.save_pickle("make_trt_module", "model.pkl", make_trt_module)
+
+
+if __name__ == "__main__":
+    main()  # pragma: no cover