holobench 1.40.0__py3-none-any.whl → 1.40.1__py3-none-any.whl

bencher/bench_cfg.py

@@ -8,15 +8,13 @@ from typing import List
 import param
 from str2bool import str2bool
 import panel as pn
-
+from datetime import datetime
 
 from bencher.variables.sweep_base import hash_sha1, describe_variable
 from bencher.variables.time import TimeSnapshot, TimeEvent
 from bencher.variables.results import OptDir
 from bencher.job import Executors
-from datetime import datetime
-
-# from bencher.results.bench_result import BenchResult
+from bencher.results.laxtex_result import to_latex
 
 
 class BenchPlotSrvCfg(param.Parameterized):
@@ -344,13 +342,32 @@ class BenchCfg(BenchRunCfg):
     def inputs_as_str(self) -> List[str]:
         return [i.name for i in self.input_vars]
 
-    def describe_sweep(self, width: int = 800, accordion=True) -> pn.pane.Markdown:
+    def to_latex(self):
+        return to_latex(self)
+
+    def describe_sweep(self, width: int = 800, accordion=True) -> pn.pane.Markdown | pn.Column:
         """Produce a markdown summary of the sweep settings"""
 
+        latex = self.to_latex()
         desc = pn.pane.Markdown(self.describe_benchmark(), width=width)
         if accordion:
-            return pn.Accordion(("Data Collection Parameters", desc))
-        return desc
+            desc = pn.Accordion(("Data Collection Parameters", desc))
+
+        sentence = self.sweep_sentence()
+        if latex is not None:
+            return pn.Column(sentence, latex, desc)
+        return pn.Column(sentence, latex, desc)
+
+    def sweep_sentence(self):
+        inputs = " by ".join([iv.name for iv in self.all_vars])
+
+        all_vars_lens = [len(iv.values()) for iv in reversed(self.all_vars)]
+        if len(all_vars_lens) == 1:
+            all_vars_lens.append(1)
+        result_sizes = "x".join([str(iv) for iv in all_vars_lens])
+        results = ", ".join([rv.name for rv in self.result_vars])
+
+        return f"Sweeping {inputs} to generate a {result_sizes} result dataframe containing {results}. "
 
     def describe_benchmark(self) -> str:
         """Generate a string summary of the inputs and results from a BenchCfg

bencher/example/meta/generate_examples.py

@@ -48,7 +48,3 @@ if __name__ == "__main__":
     convert_example_to_jupyter_notebook(
         "/workspaces/bencher/bencher/example/inputs_1D/example_1_in_2_out_repeats.py", "1D"
     )
-
-    convert_example_to_jupyter_notebook(
-        "/workspaces/bencher/bencher/example/example_simple_float.py", "1D"
-    )

bencher/example/meta/generate_meta.py

@@ -39,7 +39,7 @@ class BenchMetaGen(bch.ParametrizedSweep):
         inputs_vars_float = [
             "float1",
             "float2",
-            "float3",
+            bch.p("float3", max_level=3),
             "sigma",
         ]
 
@@ -107,10 +107,9 @@ def example_meta(run_cfg: bch.BenchRunCfg = None, report: bch.BenchReport = None
         description="""## All Combinations of Variable Sweeps and Resulting Plots
 This uses bencher to display all the combinations of plots bencher is able to produce""",
         input_vars=[
-            bch.p("float_vars", [0, 1, 2, 3]),
+            bch.p("float_vars", [0, 1]),
             "categorical_vars",
             bch.p("sample_with_repeats", [1, 20]),
-            # "sample_over_time",
         ],
         const_vars=[
             # BenchMeta.param.float_vars.with_const(1),
@@ -119,6 +118,31 @@ This uses bencher to display all the combinations of plots bencher is able to pr
             # BenchMeta.param.sample_over_time.with_const(True),
         ],
     )
+    bench.plot_sweep(
+        title="Meta Bench",
+        description="""## All Combinations of Variable Sweeps and Resulting Plots
+This uses bencher to display all the combinations of plots bencher is able to produce""",
+        input_vars=[
+            bch.p("float_vars", [2, 3]),
+            "categorical_vars",
+        ],
+    )
+
+    #     bench.plot_sweep(
+    #         title="Meta Bench",
+    #         description="""## All Combinations of Variable Sweeps and Resulting Plots
+    # This uses bencher to display all the combinations of plots bencher is able to produce""",
+    #         input_vars=[
+    #             bch.p("float_vars", [2, 3]),
+    #             "categorical_vars",
+    #         ],
+    #         const_vars=[
+    #             dict(level=3)
+    #             # BenchMeta.param.sample_with_repeats.with_const(2),
+    #             # BenchMeta.param.categorical_vars.with_const(2),
+    #             # BenchMeta.param.sample_over_time.with_const(True),
+    #         ],
+    #     )
 
     return bench
 

bencher/results/laxtex_result.py

@@ -0,0 +1,60 @@
+from __future__ import annotations
+import panel as pn
+from panel.pane import LaTeX
+from typing import Optional
+
+pn.extension("mathjax")
+
+
+def latex_text(var):
+    var_str = str(var)
+    return r"\text{" + var_str.replace("_", " ") + r"} \\"
+
+
+def input_var_to_latex(input_var):
+    vals = input_var.values()
+    displayed_vals = vals
+    if len(vals) > 5:
+        displayed_vals = [vals[i] for i in [0, 1, 0, -2, -1]]
+        displayed_vals[2] = "⋮"
+    v = r"\\ ".join([str(va) for va in displayed_vals])
+
+    latex_str = r"""\begin{array}{c}"""
+    latex_str += latex_text(input_var.name)
+    latex_str += f"{len(vals)}" + r"\times1 \\"
+    latex_str += r"\left[ \begin{array}{c}"
+    latex_str += f"{v} "
+    latex_str += r"\end{array} \right] \end{array}"
+    return latex_str
+
+
+def result_var_to_latex(bench_cfg):
+    latex_str = r"\begin{array}{c}"
+    sizes = [str(len(i.values())) for i in bench_cfg.all_vars]
+    if len(sizes) == 1:
+        sizes.insert(0, "1")
+    sizes_str = r"\times".join(reversed(sizes))
+    latex_str += sizes_str + r"\\ of \\"
+    latex_str += r" \left[\begin{array}{cc}"
+
+    for rv in bench_cfg.result_vars:
+        latex_str += latex_text(rv.name)
+    latex_str += r"\end{array} \right]"
+    latex_str += r"\end{array}"
+    return latex_str
+
+
+def to_latex(bench_cfg) -> Optional[pn.pane.LaTeX]:
+    if len(bench_cfg.all_vars) > 0:
+        latex_str = r"\["
+        for i, iv in enumerate(bench_cfg.all_vars):
+            latex_str += input_var_to_latex(iv)
+            if i != len(bench_cfg.all_vars) - 1:
+                latex_str += r"\bigtimes"
+
+        latex_str += r"\rightarrow\quad"
+        latex_str += result_var_to_latex(bench_cfg)
+        latex_str += r"\]"
+        # underscore represents subscript, so replace underscore with large space character \;
+        return LaTeX(latex_str.replace("_", r"\;"))
+    return None

{holobench-1.40.0.dist-info → holobench-1.40.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: holobench
-Version: 1.40.0
+Version: 1.40.1
 Summary: A package for benchmarking the performance of arbitrary functions
 Project-URL: Repository, https://github.com/dyson-ai/bencher
 Project-URL: Home, https://github.com/dyson-ai/bencher

{holobench-1.40.0.dist-info → holobench-1.40.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
 bencher/__init__.py,sha256=hWfQxlvuHRsFK4ZPCpRXo3nDzQB52JOUoi67wcnhopE,1890
-bencher/bench_cfg.py,sha256=gEpF1J4RAxCQPSkI8npKfhw-o3-8cw80TjiWK7As5WE,18417
+bencher/bench_cfg.py,sha256=oJ2DfZgqzY0HOYwzP_tOMLA4e4_jwRO_qsfkhcZck5E,19146
 bencher/bench_plot_server.py,sha256=nvGTr981XgWELqV7yID91j6V1UIPGtKilzxHcNWaZ6Q,4196
 bencher/bench_report.py,sha256=ikMSHceyc8cYFH-sIza167DH-H-_iiTYDm2TmusUHDc,7515
 bencher/bench_runner.py,sha256=wShmZ504BOKgHj0sOrGZtduyPfJHFFBfHRsz5tYy5_Q,7000
@@ -71,8 +71,8 @@ bencher/example/meta/example_meta.py,sha256=ZXW8LZ1DGs05RiZRHLrOx-qt9x10qL6v2tt7
 bencher/example/meta/example_meta_cat.py,sha256=j3OCi1Yra47wrXOVr8epMYPGIXjQwdQoCcCICPmE2lo,609
 bencher/example/meta/example_meta_float.py,sha256=Y-zo7QAZkpt3mQEE4guiPmRHmKjn6-y8D_AjWeGCqEs,598
 bencher/example/meta/example_meta_levels.py,sha256=ZZ14r99z6cs73ZwvBJvDKdEVIdTyHzrWsenxRgFXfXQ,1410
-bencher/example/meta/generate_examples.py,sha256=Uicdr5PlamqiFGgEzEtdPzQPfVRE6UOa1CH-3UQ6T_I,1540
-bencher/example/meta/generate_meta.py,sha256=1euRIVEAUlT3a_olmhZXbHMaG9OCSzsgKajXXwf1vOI,3874
+bencher/example/meta/generate_examples.py,sha256=EH9Rb1xgGv67F63jRcyjxVk0n56DeYHXU5cmve6G3-U,1416
+bencher/example/meta/generate_meta.py,sha256=Pu7bvu5tNKsPuEPFU90g4z1d-ZPEitqhr4rPQe_X_I8,4861
 bencher/example/optuna/example_optuna.py,sha256=0zA6IIDWe7FM0rnYJ-FHF9GZtrhYdKHQTosUD2Raw0M,2338
 bencher/example/shelved/example_float2D_scatter.py,sha256=z8ranMq8IcJ1yoVSFDncp3gw-yWG7X9lXLimXKpy5Ks,3372
 bencher/example/shelved/example_float3D_cone.py,sha256=T3dkiEhjm6z3-Vs2SjCNWPKeHk8Bp4FbANE6yXYc_YM,2970
@@ -87,6 +87,7 @@ bencher/results/dataset_result.py,sha256=qXmFMrVAo_1qM6hhV4XpQqmCz9RiVkQo6ICYmbT
 bencher/results/float_formatter.py,sha256=sX6HNCyaXdHDxC8ybVUHwCJ3qOKbPUkBOplVIHtKWjM,1746
 bencher/results/holoview_result.py,sha256=r9G9p2i7r01O2ZnsPoyOJU39ImMQhW1QvGV0mNgolc4,29463
 bencher/results/hvplot_result.py,sha256=bYSewYhPLVzW6HF_WPjAhS1ZiRp9FJHs008UEBXgH4Y,1993
+bencher/results/laxtex_result.py,sha256=EiyOzk6veaUjZyNjfhtKKqiQLoJLq9WzhZTqqepk7E8,1870
 bencher/results/optuna_result.py,sha256=QtZ4TGRun7gJoFVUjEyXKPF5yakwOSXrqEXQVJdJmm4,13587
 bencher/results/panel_result.py,sha256=lXOtfhWKSspf53Wgm94DTiVD3rliieHQW96sOdu5UYk,1336
 bencher/results/plotly_result.py,sha256=wkgfL38qJp6RviekXBYpNPeU4HCf0nbtKDAhu5QZhUg,2132
@@ -104,7 +105,7 @@ bencher/variables/results.py,sha256=Wq14e8rAj5mcK22325wcaeTMjgZ6JuduqceAHItHFY8,
 bencher/variables/sweep_base.py,sha256=gfEhKvsb16ZLbe38JewZqu0AMOHpsqwRbZbt-aCg9Bc,6258
 bencher/variables/time.py,sha256=zcRS5p4ZkFjMta9nZMEuWv86rLnPkUSqyO69QwI5q3E,3142
 resource/bencher,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-holobench-1.40.0.dist-info/METADATA,sha256=hn3E2quLA2y2qpAyWnuGBOw5lvLG85A_dpUNDc8vO10,6846
-holobench-1.40.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-holobench-1.40.0.dist-info/licenses/LICENSE,sha256=dSHXTdRY4Y7qGFMv63UksV700iff7iE-p7GGs6Sbnvo,1065
-holobench-1.40.0.dist-info/RECORD,,
+holobench-1.40.1.dist-info/METADATA,sha256=zq3FNXMLsQ3TghWMygeA0TWWR1vknTPbTjzLwoXQx_g,6846
+holobench-1.40.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+holobench-1.40.1.dist-info/licenses/LICENSE,sha256=dSHXTdRY4Y7qGFMv63UksV700iff7iE-p7GGs6Sbnvo,1065
+holobench-1.40.1.dist-info/RECORD,,