pyglove 0.4.5.dev202410230809__py3-none-any.whl → 0.4.5.dev202410280809__py3-none-any.whl

pyglove/core/object_utils/profiling.py

@@ -41,6 +41,40 @@ class TimeIt:
       """Returns whether the context has error."""
       return self.error is not None
 
+  @dataclasses.dataclass
+  class StatusSummary:
+    """Aggregated summary for repeated calls for `pg.timeit`."""
+
+    @dataclasses.dataclass
+    class Entry:
+      """Aggregated status from the `pg.timeit` calls of the same name."""
+
+      num_started: int = 0
+      num_ended: int = 0
+      num_failed: int = 0
+      avg_duration: float = 0.0
+
+      def update(self, status: 'TimeIt.Status'):
+        self.avg_duration = (
+            (self.avg_duration * self.num_started + status.elapse)
+            / (self.num_started + 1)
+        )
+        self.num_started += 1
+        if status.has_ended:
+          self.num_ended += 1
+        if status.has_error:
+          self.num_failed += 1
+
+    breakdown: dict[str, 'TimeIt.StatusSummary.Entry'] = (
+        dataclasses.field(default_factory=dict)
+    )
+
+    def aggregate(self, timeit_obj: 'TimeIt'):
+      for k, v in timeit_obj.status().items():
+        if k not in self.breakdown:
+          self.breakdown[k] = TimeIt.StatusSummary.Entry()
+        self.breakdown[k].update(v)
+
   def __init__(self, name: str):
     self._name = name
     self._start_time = None

pyglove/core/object_utils/profiling_test.py

@@ -101,6 +101,38 @@ class TimeItTest(unittest.TestCase):
     self.assertTrue(r['node.child.grandchild'].has_error)
     self.assertTrue(t2.has_error)
 
+  def test_timeit_summary(self):
+    summary = profiling.TimeIt.StatusSummary()
+    for i in range(10):
+      with profiling.timeit('node') as t:
+        time.sleep(0.1)
+        with profiling.timeit('child'):
+          time.sleep(0.1)
+          try:
+            with profiling.timeit('grandchild'):
+              time.sleep(0.1)
+              if i < 2:
+                raise ValueError('error')
+          except ValueError:
+            pass
+      summary.aggregate(t)
+    self.assertEqual(
+        list(summary.breakdown.keys()),
+        ['node', 'node.child', 'node.child.grandchild']
+    )
+    self.assertEqual(
+        [x.num_started for x in summary.breakdown.values()],
+        [10, 10, 10]
+    )
+    self.assertEqual(
+        [x.num_ended for x in summary.breakdown.values()],
+        [10, 10, 10]
+    )
+    self.assertEqual(
+        [x.num_failed for x in summary.breakdown.values()],
+        [0, 0, 2]
+    )
+
 
 if __name__ == '__main__':
   unittest.main()

{pyglove-0.4.5.dev202410230809.dist-info → pyglove-0.4.5.dev202410280809.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyglove
-Version: 0.4.5.dev202410230809
+Version: 0.4.5.dev202410280809
 Summary: PyGlove: A library for manipulating Python objects.
 Home-page: https://github.com/google/pyglove
 Author: PyGlove Authors

{pyglove-0.4.5.dev202410230809.dist-info → pyglove-0.4.5.dev202410280809.dist-info}/RECORD

@@ -62,8 +62,8 @@ pyglove/core/object_utils/json_conversion.py,sha256=1XEwz4LzEh8wHsTIC7jL7lD7L5tK
 pyglove/core/object_utils/json_conversion_test.py,sha256=KGt0r628KHi4fTfeCgrhirfINpYXIR45l5ER4z09jjI,11907
 pyglove/core/object_utils/missing.py,sha256=0liMs9iEyQxxu6UohdJ5hEM246e9Nu05qp0hqriHsl0,1459
 pyglove/core/object_utils/missing_test.py,sha256=B36p-vqUvAnXWMszAj9GOPBN0_8cq7vVF61AkcsZ9qU,1396
-pyglove/core/object_utils/profiling.py,sha256=kINgIKL2fII584A5c-dMHgQ06h0ixFCQOxE7zhLz1zE,4357
-pyglove/core/object_utils/profiling_test.py,sha256=3Rp9RBozqipJ7yMD59DGaHkLrFKrWDmxjJIDAxpHQo0,3404
+pyglove/core/object_utils/profiling.py,sha256=hNMd4BY2jCOOV3o0VTlweHxZwd8yGylDzW6GgHU1-j8,5385
+pyglove/core/object_utils/profiling_test.py,sha256=LvFy8qrAJUD-A4O7NBWZJ5YfjSGtc_zpmATCMs9hSp8,4322
 pyglove/core/object_utils/thread_local.py,sha256=i-CnyY3VREtLfAj4_JndBnsKuQLIgwG29ma8dAyRxbI,4839
 pyglove/core/object_utils/thread_local_test.py,sha256=EvU1-TF7KqpLQxxBvHd7dxtuY22YUQSIwQ0UcR-NORA,6816
 pyglove/core/object_utils/value_location.py,sha256=lSFQNTazY2M6_nRLmMbouqZAcZSiOLZQnmQPMD2FDMs,26770
@@ -184,8 +184,8 @@ pyglove/ext/scalars/randoms.py,sha256=LkMIIx7lOq_lvJvVS3BrgWGuWl7Pi91-lA-O8x_gZs
 pyglove/ext/scalars/randoms_test.py,sha256=nEhiqarg8l_5EOucp59CYrpO2uKxS1pe0hmBdZUzRNM,2000
 pyglove/ext/scalars/step_wise.py,sha256=IDw3tuTpv0KVh7AN44W43zqm1-E0HWPUlytWOQC9w3Y,3789
 pyglove/ext/scalars/step_wise_test.py,sha256=TL1vJ19xVx2t5HKuyIzGoogF7N3Rm8YhLE6JF7i0iy8,2540
-pyglove-0.4.5.dev202410230809.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-pyglove-0.4.5.dev202410230809.dist-info/METADATA,sha256=Y17S_a9GcDhytQb65ieOyUu9YP6zgRaNh-dkrXrTtgk,6666
-pyglove-0.4.5.dev202410230809.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-pyglove-0.4.5.dev202410230809.dist-info/top_level.txt,sha256=wITzJSKcj8GZUkbq-MvUQnFadkiuAv_qv5qQMw0fIow,8
-pyglove-0.4.5.dev202410230809.dist-info/RECORD,,
+pyglove-0.4.5.dev202410280809.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+pyglove-0.4.5.dev202410280809.dist-info/METADATA,sha256=WudMdo1hklsd4DOJlo6tlV8ABP_HfbZYIh9B3ABI7ac,6666
+pyglove-0.4.5.dev202410280809.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+pyglove-0.4.5.dev202410280809.dist-info/top_level.txt,sha256=wITzJSKcj8GZUkbq-MvUQnFadkiuAv_qv5qQMw0fIow,8
+pyglove-0.4.5.dev202410280809.dist-info/RECORD,,