halib 0.1.81__tar.gz → 0.1.83__tar.gz

{halib-0.1.81 → halib-0.1.83}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: halib
-Version: 0.1.81
+Version: 0.1.83
 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.81**
+**Version 0.1.83**
 
 + `research/profiler`: add `zProfiler` class to measure execution time of contexts and steps, with support for dynamic color scales in plots.
 

{halib-0.1.81 → halib-0.1.83}/README.md

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

{halib-0.1.81 → halib-0.1.83}/halib/research/profiler.py

@@ -95,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():
@@ -170,7 +170,7 @@ class zProfiler:
     @classmethod
     @classmethod
     def plot_formatted_data(
-        cls, profiler_data, outdir=None, file_format="png", do_show=False
+        cls, profiler_data, outdir=None, file_format="png", do_show=False, tag=""
     ):
         """
         Plot each context in a separate figure with bar + pie charts.
@@ -194,10 +194,11 @@ class zProfiler:
             # 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))
 
@@ -237,10 +238,10 @@ class zProfiler:
                 row=1,
                 col=2,
             )
-
+            tag_str = f"[{tag}] " if tag else ""
             # Layout
             fig.update_layout(
-                title_text=f"Context Profiler: {ctx}",
+                title_text=f"[{tag_str}Context Profiler: {ctx}",
                 width=1000,
                 height=400,
                 showlegend=True,
@@ -257,7 +258,8 @@ class zProfiler:
 
             # Save figure
             if outdir is not None:
-                file_path = os.path.join(outdir, f"{ctx}_summary.{file_format.lower()}")
+                file_prefix = {ctx} if len(tag_str) == 0 else f"{tag_str}_{ctx}"
+                file_path = os.path.join(outdir, f"{file_prefix}_summary.{file_format.lower()}")
                 fig.write_image(file_path)
                 print(f"Saved figure: {file_path}")
 
@@ -265,7 +267,7 @@ class zProfiler:
 
         return results
 
-    def report_and_plot(self, outdir=None, file_format="png", do_show=False):
+    def report_and_plot(self, outdir=None, file_format="png", do_show=False, tag=""):
         """
         Generate the profiling report and plot the formatted data.
 
@@ -277,7 +279,7 @@ class zProfiler:
         report = self.get_report_dict()
         self.get_report_dict(with_detail=False)
         return self.plot_formatted_data(
-            report, outdir=outdir, file_format=file_format, do_show=do_show
+            report, outdir=outdir, file_format=file_format, do_show=do_show, tag=tag
         )
     def meta_info(self):
         """

{halib-0.1.81 → halib-0.1.83}/halib.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: halib
-Version: 0.1.81
+Version: 0.1.83
 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.81**
+**Version 0.1.83**
 
 + `research/profiler`: add `zProfiler` class to measure execution time of contexts and steps, with support for dynamic color scales in plots.
 

{halib-0.1.81 → halib-0.1.83}/setup.py

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