From 055676de4b02ef399f8c13a23cff6e901eb7d301 Mon Sep 17 00:00:00 2001
From: generatedunixname89002005287564
 <generatedunixname89002005287564@meta.com>
Date: Wed, 15 Nov 2023 06:16:50 -0800
Subject: [PATCH] multipy

Reviewed By: zsol

Differential Revision: D51343327

fbshipit-source-id: 83b7a51a916c6f4949a98d5586e2ca3d7ee93b49
---
 multipy/runtime/example/generate_examples.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/multipy/runtime/example/generate_examples.py b/multipy/runtime/example/generate_examples.py
index f947f3e1..fc2acdb7 100644
--- 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