pydftracer 2.0.5.post1.dev0__tar.gz → 2.0.5.post2.dev0__tar.gz

{pydftracer-2.0.5.post1.dev0 → pydftracer-2.0.5.post2.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydftracer
-Version: 2.0.5.post1.dev0
+Version: 2.0.5.post2.dev0
 Summary: Python libs for DFTracer
 Author-email: "Hariharan Devarajan (Hari)" <hariharandev1@llnl.gov>, Ray Andrew Sinurat <raydreww@gmail.com>
 Maintainer-email: "Hariharan Devarajan (Hari)" <hariharandev1@llnl.gov>, Ray Andrew Sinurat <raydreww@gmail.com>

{pydftracer-2.0.5.post1.dev0 → pydftracer-2.0.5.post2.dev0}/python/dftracer/python/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '2.0.5.post1.dev0'
-__version_tuple__ = version_tuple = (2, 0, 5, 'post1', 'dev0')
+__version__ = version = '2.0.5.post2.dev0'
+__version_tuple__ = version_tuple = (2, 0, 5, 'post2', 'dev0')
 
-__commit_id__ = commit_id = 'g676e7a176'
+__commit_id__ = commit_id = 'gb88b10fd4'

{pydftracer-2.0.5.post1.dev0 → pydftracer-2.0.5.post2.dev0}/python/dftracer/python/ai_common.py

@@ -300,51 +300,57 @@ class _DFTracerAI:
         iter_val = 1
 
         start: int = 0
+        end: int = 0
+        t0: int = 0
+        scope_open = False
         if DFTRACER_ENABLE and self.profiler._enable:
             dftracer.get_instance().enter_event()
+            scope_open = True
             start = dftracer.get_instance().get_time()
 
-        for v in iterator:
-            if DFTRACER_ENABLE and self.profiler._enable:
-                end = dftracer.get_instance().get_time()
-                self.profiler._arguments_int[ITER_COUNT_NAME] = TagValue(
-                    iter_val, TagDType.INT, TagType.KEY
-                ).value()
-                if include_iter:
-                    dftracer.get_instance().log_event(
-                        name=iter_name,
-                        cat=self.profiler._cat,
-                        start_time=start,
-                        duration=end - start,
-                        int_args=self.profiler._arguments_int,
-                        float_args=self.profiler._arguments_float,
-                        string_args=self.profiler._arguments_string,
-                    )
-                dftracer.get_instance().exit_event()
-                dftracer.get_instance().enter_event()
-                t0 = dftracer.get_instance().get_time()
-
-            yield v
-
-            if DFTRACER_ENABLE and self.profiler._enable:
-                t1 = dftracer.get_instance().get_time()
-                if include_block:
-                    dftracer.get_instance().log_event(
-                        name=block_name,
-                        cat=self.profiler._cat,
-                        start_time=t0,
-                        duration=t1 - t0,
-                        int_args=self.profiler._arguments_int,
-                        float_args=self.profiler._arguments_float,
-                        string_args=self.profiler._arguments_string,
-                    )
-                dftracer.get_instance().exit_event()
+        try:
+            for v in iterator:
+                if DFTRACER_ENABLE and self.profiler._enable:
+                    end = dftracer.get_instance().get_time()
+                    dftracer.get_instance().exit_event()
+                    dftracer.get_instance().enter_event()
+                    t0 = dftracer.get_instance().get_time()
 
-                iter_val += 1
-                dftracer.get_instance().enter_event()
-                start = dftracer.get_instance().get_time()
-        if DFTRACER_ENABLE and self.profiler._enable:
-            dftracer.get_instance().exit_event()
+                yield v
+
+                if DFTRACER_ENABLE and self.profiler._enable:
+                    t1 = dftracer.get_instance().get_time()
+                    self.profiler._arguments_int[ITER_COUNT_NAME] = TagValue(
+                        iter_val, TagDType.INT, TagType.KEY
+                    ).value()
+                    if include_iter:
+                        dftracer.get_instance().log_event(
+                            name=iter_name,
+                            cat=self.profiler._cat,
+                            start_time=start,
+                            duration=end - start,
+                            int_args=self.profiler._arguments_int,
+                            float_args=self.profiler._arguments_float,
+                            string_args=self.profiler._arguments_string,
+                        )
+                    if include_block:
+                        dftracer.get_instance().log_event(
+                            name=block_name,
+                            cat=self.profiler._cat,
+                            start_time=t0,
+                            duration=t1 - t0,
+                            int_args=self.profiler._arguments_int,
+                            float_args=self.profiler._arguments_float,
+                            string_args=self.profiler._arguments_string,
+                        )
+                    dftracer.get_instance().exit_event()
+
+                    iter_val += 1
+                    dftracer.get_instance().enter_event()
+                    start = dftracer.get_instance().get_time()
+        finally:
+            if scope_open:
+                dftracer.get_instance().exit_event()
 
 
 class DFTracerAI(_DFTracerAI):

{pydftracer-2.0.5.post1.dev0 → pydftracer-2.0.5.post2.dev0}/python/pydftracer.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydftracer
-Version: 2.0.5.post1.dev0
+Version: 2.0.5.post2.dev0
 Summary: Python libs for DFTracer
 Author-email: "Hariharan Devarajan (Hari)" <hariharandev1@llnl.gov>, Ray Andrew Sinurat <raydreww@gmail.com>
 Maintainer-email: "Hariharan Devarajan (Hari)" <hariharandev1@llnl.gov>, Ray Andrew Sinurat <raydreww@gmail.com>

pydftracer-2.0.5.post2.dev0/python/pydftracer.egg-info/scm_version.json

@@ -0,0 +1,8 @@
+{
+  "tag": "2.0.5",
+  "distance": 2,
+  "node": "gb88b10fd4221cd6ed3653a3c34c1eab883bce041",
+  "dirty": false,
+  "branch": "develop",
+  "node_date": "2026-10-14"
+}

pydftracer-2.0.5.post1.dev0/python/pydftracer.egg-info/scm_version.json

@@ -1,8 +0,0 @@
-{
-  "tag": "2.0.5",
-  "distance": 1,
-  "node": "g676e7a176e5ecd4005db1ebc0b2a003658a8a8ff",
-  "dirty": false,
-  "branch": "develop",
-  "node_date": "2026-10-02"
-}