hbt (#184)

Summary: Pull Request resolved: #184 Reviewed By: zsol Differential Revision: D51104970 fbshipit-source-id: 7bee9e8b5eb9ecebf4343a8a519d540740c6110f
facebookincubator · Nov 8, 2023 · ebe968c · ebe968c
1 parent df570fb
commit ebe968c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/hbt/src/intel_pt/main.py b/hbt/src/intel_pt/main.py
@@ -10,7 +10,7 @@
 from hbt.intel_pt.tracer import Tracer
 
 
-if __name__ == "__main__":
+def main() -> None:
     parser = argparse.ArgumentParser(description="Intel PT tracer.")
     parser.add_argument("pid", type=int, default=None, help="PIDs to filter on.")
     parser.add_argument(
@@ -32,3 +32,7 @@
         for p in t.read():
             print(f"Event at iteration {i}: {p}")
     t.stop()
+
+
+if __name__ == "__main__":
+    main()  # pragma: no cover