halib 0.1.53__py3-none-any.whl → 0.1.54__py3-none-any.whl

halib/research/perftb.py

@@ -12,10 +12,10 @@ from typing import Dict, List, Union, Optional
 from rich.pretty import pprint
 import pandas as pd
 
-# from halib import *
+from halib import *
 # internal imports
-from ..filetype import csvfile
-from ..common import ConsoleLog
+# from ..filetype import csvfile
+# from ..common import ConsoleLog
 
 
 class DatasetMetrics:
@@ -249,12 +249,15 @@ class PerfTB:
 
         current_our_method = -1  # Start with -1 to avoid index error
         exp_pattern_dict = {}
+        shown_legends = set()
         for row_idx, metric in enumerate(metric_list, start=1):
             metric_df = df[df["Metric"] == metric]
             list_exp = list(metric_df["Experiment"].unique())
             if custom_sort_exp_fn:
                 list_exp = custom_sort_exp_fn(list_exp)
             for exp in list_exp:
+                showlegend = exp not in shown_legends
+                shown_legends.add(exp) # since it is a set, it will only keep unique values
                 should_highlight = (
                     custom_highlight_method_fn is not None and custom_highlight_method_fn(exp)
                 )
@@ -274,7 +277,7 @@ class PerfTB:
                         y=exp_df["Value"],
                         name=f"{exp}",
                         legendgroup=exp,
-                        showlegend=(row_idx == 1),  # Show legend only for the first row
+                        showlegend=showlegend,  # Show legend only for the first row
                         marker=dict(
                             color=color_map[exp],
                             pattern=(
@@ -749,8 +752,19 @@ def test_mics() -> None:
         custom_sort_exp_fn=lambda exps: sorted(exps, reverse=True),
         open_plot=True,
     )
+def test_bench2():
+    perftb = PerfTB.from_csv(
+        "test/bench2.csv",
+        sep=";")
+    perftb.display()
+    perftb.plot(
+        save_path="zout/bench2_plot.svg",
+        title="Bench2 Performance Comparison",
+        open_plot=True,
+    )
 
 
 # Example usage
 if __name__ == "__main__":
-    test_mics()
+    # test_mics()
+    test_bench2()

{halib-0.1.53.dist-info → halib-0.1.54.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: halib
-Version: 0.1.53
+Version: 0.1.54
 Summary: Small library for common tasks
 Author: Hoang Van Ha
 Author-email: hoangvanhauit@gmail.com
@@ -45,7 +45,7 @@ Requires-Dist: dataclass-wizard
 
 Helper package for coding and automation
 
-**Version 0.1.53**
+**Version 0.1.54**
 
 + add `util/dataclass_util` to help dynamically create `dataclass` classes from dictionary or YAML file, including support for nested dataclasses. From there, we can use `dataclass_wizard` to create a list of `dataclass` classes with the help from ChatGPT.
 

{halib-0.1.53.dist-info → halib-0.1.54.dist-info}/RECORD

@@ -30,7 +30,7 @@ halib/online/projectmake.py,sha256=Zrs96WgXvO4nIrwxnCOletL4aTBge-EoF0r7hpKO1w8,4
 halib/research/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 halib/research/benchquery.py,sha256=FuKnbWQtCEoRRtJAfN-zaN-jPiO_EzsakmTOMiqi7GQ,4626
 halib/research/dataset.py,sha256=QU0Hr5QFb8_XlvnOMgC9QJGIpwXAZ9lDd0RdQi_QRec,6743
-halib/research/perftb.py,sha256=BEwWIteVGtXtUkfboLYbIENxYL1vpVIVioU9eLrbqic,30007
+halib/research/perftb.py,sha256=Fj1l3RFGrswR7UzmMDafRwhQkInUUcNfxt5Wayc2swc,30465
 halib/research/plot.py,sha256=-pDUk4z3C_GnyJ5zWmf-mGMdT4gaipVJWzIgcpIPiRk,9448
 halib/research/torchloader.py,sha256=yqUjcSiME6H5W210363HyRUrOi3ISpUFAFkTr1w4DCw,6503
 halib/research/wandb_op.py,sha256=YzLEqME5kIRxi3VvjFkW83wnFrsn92oYeqYuNwtYRkY,4188
@@ -44,8 +44,8 @@ halib/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 halib/utils/dataclass_util.py,sha256=aD5Ik9BRJzIEyfdlOCHmOddA7TM_TIAfapZfU0vMigE,1414
 halib/utils/listop.py,sha256=Vpa8_2fI0wySpB2-8sfTBkyi_A4FhoFVVvFiuvW8N64,339
 halib/utils/tele_noti.py,sha256=-4WXZelCA4W9BroapkRyIdUu9cUVrcJJhegnMs_WpGU,5928
-halib-0.1.53.dist-info/LICENSE.txt,sha256=qZssdna4aETiR8znYsShUjidu-U4jUT9Q-EWNlZ9yBQ,1100
-halib-0.1.53.dist-info/METADATA,sha256=TgyMGbS3YvF5Nm7LzNvbAZBp6P4Oc0Tk3cGQY4c6G4I,4675
-halib-0.1.53.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-halib-0.1.53.dist-info/top_level.txt,sha256=7AD6PLaQTreE0Fn44mdZsoHBe_Zdd7GUmjsWPyQ7I-k,6
-halib-0.1.53.dist-info/RECORD,,
+halib-0.1.54.dist-info/LICENSE.txt,sha256=qZssdna4aETiR8znYsShUjidu-U4jUT9Q-EWNlZ9yBQ,1100
+halib-0.1.54.dist-info/METADATA,sha256=WzE-_Dwk2TFEWoTPVAHQLwTDOlV7BXaDvG1CkBP5d3E,4675
+halib-0.1.54.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+halib-0.1.54.dist-info/top_level.txt,sha256=7AD6PLaQTreE0Fn44mdZsoHBe_Zdd7GUmjsWPyQ7I-k,6
+halib-0.1.54.dist-info/RECORD,,