halib 0.1.82__py3-none-any.whl → 0.1.84__py3-none-any.whl

halib/research/profiler.py

@@ -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.
@@ -238,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,
@@ -258,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}")
 
@@ -266,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.
 
@@ -278,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.82.dist-info → halib-0.1.84.dist-info}/METADATA

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

{halib-0.1.82.dist-info → halib-0.1.84.dist-info}/RECORD

@@ -36,7 +36,7 @@ halib/research/metrics.py,sha256=Xgv0GUGo-o-RJaBOmkRCRpQJaYijF_1xeKkyYU_Bv4U,524
 halib/research/perfcalc.py,sha256=qDa0sqfpWrwGZVJtjuUVFK7JX6j8xyXP9OnnfYmdamg,15898
 halib/research/perftb.py,sha256=FWg0b8wSgy4UwuvHSXwEqvTq1Rhi-z-HtAKuQg1lWc4,30989
 halib/research/plot.py,sha256=-pDUk4z3C_GnyJ5zWmf-mGMdT4gaipVJWzIgcpIPiRk,9448
-halib/research/profiler.py,sha256=xlVAJ8mgPtC-Y1J-avuSZBloWFJgcvAjmTeOYjckvS8,11588
+halib/research/profiler.py,sha256=IW__mJbOypes_Vl5VNVB9YQS-xIpLJN_l7iuzZyK_q8,11761
 halib/research/torchloader.py,sha256=yqUjcSiME6H5W210363HyRUrOi3ISpUFAFkTr1w4DCw,6503
 halib/research/wandb_op.py,sha256=YzLEqME5kIRxi3VvjFkW83wnFrsn92oYeqYuNwtYRkY,4188
 halib/sys/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -52,8 +52,8 @@ halib/utils/gpu_mon.py,sha256=vD41_ZnmPLKguuq9X44SB_vwd9JrblO4BDzHLXZhhFY,2233
 halib/utils/listop.py,sha256=Vpa8_2fI0wySpB2-8sfTBkyi_A4FhoFVVvFiuvW8N64,339
 halib/utils/tele_noti.py,sha256=-4WXZelCA4W9BroapkRyIdUu9cUVrcJJhegnMs_WpGU,5928
 halib/utils/video.py,sha256=ZqzNVPgc1RZr_T0OlHvZ6SzyBpL7O27LtB86JMbBuR0,3059
-halib-0.1.82.dist-info/licenses/LICENSE.txt,sha256=qZssdna4aETiR8znYsShUjidu-U4jUT9Q-EWNlZ9yBQ,1100
-halib-0.1.82.dist-info/METADATA,sha256=FmbjEQLCE3jqJT6H2uyn6MfyZXsFhJwO80l93Be_zsE,5864
-halib-0.1.82.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-halib-0.1.82.dist-info/top_level.txt,sha256=7AD6PLaQTreE0Fn44mdZsoHBe_Zdd7GUmjsWPyQ7I-k,6
-halib-0.1.82.dist-info/RECORD,,
+halib-0.1.84.dist-info/licenses/LICENSE.txt,sha256=qZssdna4aETiR8znYsShUjidu-U4jUT9Q-EWNlZ9yBQ,1100
+halib-0.1.84.dist-info/METADATA,sha256=7rx1qsP50k17bZ_K9ffKdpBR3fLUtKpmYu-Q-xZTf8Q,5864
+halib-0.1.84.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+halib-0.1.84.dist-info/top_level.txt,sha256=7AD6PLaQTreE0Fn44mdZsoHBe_Zdd7GUmjsWPyQ7I-k,6
+halib-0.1.84.dist-info/RECORD,,