From 1f92c4e9d25121326d77838035748fc7b71a5cfd Mon Sep 17 00:00:00 2001
From: generatedunixname89002005287564
 <generatedunixname89002005287564@meta.com>
Date: Wed, 31 Jan 2024 07:43:49 -0800
Subject: [PATCH] vision/fair

Reviewed By: zsol

Differential Revision: D53258682

fbshipit-source-id: 3f006b5f31a2b1ffdc6323d3a3b08ac46c3162ce
---
 tests/benchmarks/bm_main.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/tests/benchmarks/bm_main.py b/tests/benchmarks/bm_main.py
index 88e0585e..ce10012e 100755
--- a/tests/benchmarks/bm_main.py
+++ b/tests/benchmarks/bm_main.py
@@ -12,7 +12,7 @@ import sys
 from os.path import dirname, isfile, join
 
 
-if __name__ == "__main__":
+def main() -> None:
     # pyre-ignore[16]
     if len(sys.argv) > 1:
         # Parse from flags.
@@ -36,3 +36,7 @@ if __name__ == "__main__":
     os.environ["PYTHONPATH"] = ":".join(sys.path)
     for file_name in file_names:
         subprocess.check_call([sys.executable, file_name])
+
+
+if __name__ == "__main__":
+    main()  # pragma: no cover