Use context manager to decrease memory load in DQM (#174)

cta-observatory · Feb 5, 2025 · 12a8133 · 12a8133
1 parent fa2fed9
commit 12a8133
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/src/nectarchain/dqm/start_dqm.py b/src/nectarchain/dqm/start_dqm.py
@@ -238,13 +238,16 @@ def CreateFigFolder(name, type):
         path2 = f"{NectarPath}/runs/{arg}"
         print(path2)
 
-        reader = EventSource(input_url=path2, config=config, max_events=args.max_events)
-
-        for evt in tqdm(
-            reader, total=args.max_events if args.max_events else len(reader), unit="ev"
-        ):
-            for p in processors:
-                p.ProcessEvent(evt, noped)
+        with EventSource(
+            input_url=path2, config=config, max_events=args.max_events
+        ) as reader:
+            for evt in tqdm(
+                reader,
+                total=args.max_events if args.max_events else len(reader),
+                unit="ev",
+            ):
+                for p in processors:
+                    p.ProcessEvent(evt, noped)
 
     for p in processors:
         p.FinishRun()