halib 0.1.80__tar.gz → 0.1.82__tar.gz

{halib-0.1.80 → halib-0.1.82}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: halib
-Version: 0.1.80
+Version: 0.1.82
 Summary: Small library for common tasks
 Author: Hoang Van Ha
 Author-email: hoangvanhauit@gmail.com
@@ -52,7 +52,7 @@ Dynamic: summary
 
 Helper package for coding and automation
 
-**Version 0.1.80**
+**Version 0.1.82**
 
 + `research/profiler`: add `zProfiler` class to measure execution time of contexts and steps, with support for dynamic color scales in plots.
 

{halib-0.1.80 → halib-0.1.82}/README.md

@@ -1,6 +1,6 @@
 Helper package for coding and automation
 
-**Version 0.1.80**
+**Version 0.1.82**
 
 + `research/profiler`: add `zProfiler` class to measure execution time of contexts and steps, with support for dynamic color scales in plots.
 

{halib-0.1.80 → halib-0.1.82}/halib/research/profiler.py

@@ -9,6 +9,7 @@ from threading import Lock
 from plotly.subplots import make_subplots
 import plotly.graph_objects as go
 import plotly.express as px  # for dynamic color scales
+from ..common import ConsoleLog
 
 from loguru import logger
 
@@ -94,7 +95,7 @@ class zProfiler:
         │   }
         """
         assert (
-            len(ctx_step_dict.keys()) > 1
+            len(ctx_step_dict.keys()) > 0
         ), "step_dict must have only one key (step_name) for detail."
 
         for step_name, time_list in ctx_step_dict.items():
@@ -190,13 +191,14 @@ class zProfiler:
             percent_times = summary["percent_time"]
 
             step_names = [s.replace("avg_", "") for s in avg_times.keys()]
-            pprint(f'{step_names=}')
+            # pprint(f'{step_names=}')
             n_steps = len(step_names)
 
+            assert n_steps > 0, "No steps found for context: {}".format(ctx)
             # Generate dynamic colors
             colors = px.colors.sample_colorscale(
                 "Viridis", [i / (n_steps - 1) for i in range(n_steps)]
-            )
+            ) if n_steps > 1 else [px.colors.sample_colorscale("Viridis", [0])[0]]
             # pprint(f'{len(colors)} colors generated for {n_steps} steps')
             color_map = dict(zip(step_names, colors))
 
@@ -278,6 +280,16 @@ class zProfiler:
         return self.plot_formatted_data(
             report, outdir=outdir, file_format=file_format, do_show=do_show
         )
+    def meta_info(self):
+        """
+        Print the structure of the profiler's time dictionary.
+        Useful for debugging and understanding the profiler's internal state.
+        """
+        for ctx_name, ctx_dict in self.time_dict.items():
+            with ConsoleLog(f"Context: {ctx_name}"):
+                step_names = list(ctx_dict['step_dict'].keys())
+                for step_name in step_names:
+                    pprint(f"Step: {step_name}")
 
     def save_report_dict(self, output_file, with_detail=False):
         try:

{halib-0.1.80 → halib-0.1.82}/halib.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: halib
-Version: 0.1.80
+Version: 0.1.82
 Summary: Small library for common tasks
 Author: Hoang Van Ha
 Author-email: hoangvanhauit@gmail.com
@@ -52,7 +52,7 @@ Dynamic: summary
 
 Helper package for coding and automation
 
-**Version 0.1.80**
+**Version 0.1.82**
 
 + `research/profiler`: add `zProfiler` class to measure execution time of contexts and steps, with support for dynamic color scales in plots.
 

{halib-0.1.80 → halib-0.1.82}/setup.py

@@ -8,7 +8,7 @@ with open("requirements.txt") as f:
 
 setuptools.setup(
     name="halib",
-    version="0.1.80",
+    version="0.1.82",
     author="Hoang Van Ha",
     author_email="hoangvanhauit@gmail.com",
     description="Small library for common tasks",