Import argparse and sys only when necessary

key4hep · Sep 6, 2024 · 4cb4882 · 4cb4882
1 parent d883551
commit 4cb4882
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/scripts/createEDM4hepFile.py b/scripts/createEDM4hepFile.py
@@ -6,8 +6,6 @@
 import podio
 import edm4hep
 from itertools import count
-import argparse
-import sys
 
 FRAMES = 3  # How many frames or events will be written
 VECTORSIZE = 5  # For vector members, each vector member will have this size
@@ -472,6 +470,9 @@ def create_frame():
 
 
 if __name__ == "__main__":
+    import argparse
+    import sys
+
     parser = argparse.ArgumentParser(description="Create a file with EDM4hep data")
     parser.add_argument(
         "--rntuple", action="store_true", help="Use a ROOT ntuple instead of EDM4hep"