holobench 1.3.1__py3-none-any.whl → 1.3.2__py3-none-any.whl

bencher/__init__.py

@@ -18,7 +18,14 @@ from .variables.results import (
     curve,
 )
 from .plotting.plot_filter import VarRange, PlotFilter
-from .utils import hmap_canonical_input, get_nearest_coords, make_namedtuple
+from .utils import (
+    hmap_canonical_input,
+    get_nearest_coords,
+    make_namedtuple,
+    gen_path,
+    gen_image_path,
+    gen_video_path,
+)
 from .variables.parametrised_sweep import ParametrizedSweep
 from .caching import CachedParams
 from .results.bench_result import BenchResult
@@ -26,3 +33,4 @@ from .results.panel_result import PanelResult
 from .results.holoview_result import ReduceType, HoloviewResult
 from .bench_report import BenchReport
 from .job import Executors
+from .video_writer import VideoWriter

bencher/bencher.py

@@ -228,7 +228,7 @@ class Bench(BenchPlotServer):
                 )
                 input_vars = self.worker_class_instance.get_inputs_only()
                 for i in input_vars:
-                    logging.info(i.name)
+                    logging.info(f"input var: {i.name}")
             if result_vars is None:
                 logging.info(
                     "No results variables passed, using all result variables in bench class:"

bencher/example/example_image.py

@@ -25,7 +25,7 @@ class BenchPolygons(bch.ParametrizedSweep):
         self.update_params_from_kwargs(**kwargs)
         points = polygon_points(self.radius, self.sides)
         self.hmap = hv.Curve(points)
-        self.polygon = self.points_to_polygon_png(points, self.gen_image_path("polygon"))
+        self.polygon = self.points_to_polygon_png(points, bch.gen_image_path("polygon"))
         return super().__call__()
 
     def points_to_polygon_png(self, points: list[float], filename: str):

bencher/example/example_video.py

@@ -66,7 +66,7 @@ class TuringPattern(bch.ParametrizedSweep):
                 artists.append([ax.imshow(U)])
 
         ani = ArtistAnimation(fig=fig, artists=artists, interval=60, blit=True)
-        path = self.gen_path("turing", "vid", ".mp4")
+        path = bch.gen_video_path("turing", ".mp4")
         # path = self.gen_path("turing", "vid", ".gif")
         print(f"saving {len(artists)} frames")
         ani.save(path)

bencher/results/holoview_result.py

@@ -388,7 +388,6 @@ class HoloviewResult(PanelResult):
         return htmap + tap_htmap
 
     def to_nd_layout(self, hmap_name: str) -> hv.NdLayout:
-        print(self.bench_cfg.hmap_kdims)
         return hv.NdLayout(self.get_hmap(hmap_name), kdims=self.bench_cfg.hmap_kdims).opts(
             shared_axes=False, shared_datasource=False
         )

bencher/results/panel_result.py

@@ -1,4 +1,5 @@
 from typing import Optional, Any
+from pathlib import Path
 from functools import partial
 import panel as pn
 import xarray as xr
@@ -23,10 +24,12 @@ class PanelResult(BenchResultBase):
             xr_dataset = self.to_hv_dataset(ReduceType.SQUEEZE)
 
             def to_video_da(da, **kwargs):
-                vid = pn.pane.Video(da, autoplay=True, **kwargs)
-                vid.loop = True
-                vid_p.append(vid)
-                return vid
+                if da is not None and Path(da).exists():
+                    vid = pn.pane.Video(da, autoplay=True, **kwargs)
+                    vid.loop = True
+                    vid_p.append(vid)
+                    return vid
+                return pn.pane.Markdown(f"video does not exist {da}")
 
             plot_callback = partial(self.ds_to_container, container=partial(to_video_da, **kwargs))
 

bencher/utils.py

@@ -5,6 +5,8 @@ import hashlib
 import re
 import math
 from colorsys import hsv_to_rgb
+from pathlib import Path
+from uuid import uuid4
 
 
 def hmap_canonical_input(dic: dict) -> tuple:
@@ -106,3 +108,17 @@ def int_to_col(int_val, sat=0.5, val=0.95, alpha=-1) -> tuple[float, float, floa
 
 def color_tuple_to_css(color: tuple[float, float, float]) -> str:
     return f"rgb{(color[0] * 255, color[1] * 255, color[2] * 255)}"
+
+
+def gen_path(filename, folder, suffix):
+    path = Path(f"cachedir/{folder}/{filename}/")
+    path.mkdir(parents=True, exist_ok=True)
+    return f"{path.absolute().as_posix()}/{filename}_{uuid4()}{suffix}"
+
+
+def gen_video_path(video_name: str, extension: str = ".webm") -> str:
+    return gen_path(video_name, "vid", extension)
+
+
+def gen_image_path(image_name: str, filetype=".png") -> str:
+    return gen_path(image_name, "img", filetype)

bencher/variables/inputs.py

@@ -185,3 +185,9 @@ class FloatSweep(Number, SweepBase):
             ]
             return [self.sample_values[i] for i in indices]
         return self.sample_values
+
+
+def box(name, center, width):
+    var = FloatSweep(default=center, bounds=(center - width, center + width))
+    var.name = name
+    return var

bencher/variables/parametrised_sweep.py

@@ -3,7 +3,7 @@ from typing import List, Tuple, Any
 from param import Parameter, Parameterized
 import holoviews as hv
 import panel as pn
-from pathlib import Path
+
 
 from bencher.utils import make_namedtuple, hash_sha1
 from bencher.variables.results import (
@@ -17,8 +17,6 @@ from bencher.variables.results import (
     ResultReference,
 )
 
-from uuid import uuid4
-
 
 class ParametrizedSweep(Parameterized):
     """Parent class for all Sweep types that need a custom hash"""
@@ -197,23 +195,12 @@ class ParametrizedSweep(Parameterized):
     def plot_hmap(self, **kwargs):
         return self.__call__(**kwargs)["hmap"]
 
-    def gen_path(self, filename, folder, suffix):
-        path = Path(f"cachedir/{folder}") / Path(filename)
-        path.mkdir(parents=True, exist_ok=True)
-        return f"{path.absolute().as_posix()}_{uuid4()}{suffix}"
-
-    def gen_video_path(self, video_name: str) -> str:
-        return self.gen_path(video_name, "vid", ".webm")
-
-    def gen_image_path(self, image_name: str, filetype=".png") -> str:
-        return self.gen_path(image_name, "img", filetype)
-
     def to_bench(self, run_cfg=None, report=None, name: str = None):
         from bencher import Bench
 
         assert isinstance(self, ParametrizedSweep)
 
         if name is None:
-            name = self.name
+            name = self.name[:-5]  # param adds 5 digit number to the end, so remove it
 
         return Bench(name, self, run_cfg=run_cfg, report=report)

bencher/video_writer.py

@@ -0,0 +1,17 @@
+from bencher import gen_video_path
+
+
+class VideoWriter:
+    def __init__(self, filename: str) -> None:
+        self.images = []
+        self.filename = gen_video_path(filename)
+
+    def append(self, img):
+        self.images.append(img)
+
+    def write(self, bitrate: int = 1500) -> str:
+        import moviepy.video.io.ImageSequenceClip
+
+        clip = moviepy.video.io.ImageSequenceClip.ImageSequenceClip(self.images, fps=30)
+        clip.write_videofile(self.filename, bitrate=f"{bitrate}k")
+        return self.filename

{holobench-1.3.1.dist-info → holobench-1.3.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: holobench
-Version: 1.3.1
+Version: 1.3.2
 Summary: A package for benchmarking the performance of arbitrary functions
 Author-email: Austin Gregg-Smith <blooop@gmail.com>
 Requires-Python: >=3.10
@@ -22,6 +22,7 @@ Requires-Dist: scikit-learn>=1.2,<=1.3.2
 Requires-Dist: str2bool>=1.1,<=1.1
 Requires-Dist: scoop>=0.7.0,<=0.7.2.0
 Requires-Dist: ffmpeg-downloader>=0.3.0,<=0.3.0
+Requires-Dist: moviepy>=1.0.3,<=1.0.3
 Requires-Dist: black>=23,<=23.12.1 ; extra == "test"
 Requires-Dist: pylint>=2.16,<=3.0.3 ; extra == "test"
 Requires-Dist: pytest-cov>=4.1,<=4.1 ; extra == "test"

{holobench-1.3.1.dist-info → holobench-1.3.2.dist-info}/RECORD

@@ -1,13 +1,14 @@
-bencher/__init__.py,sha256=wfTebGwGL5ImpwkFife9xuUxhH4vSHRewyUzXVEu8g4,1052
+bencher/__init__.py,sha256=l3s3a3hf7E6RciydXdHNuzzXPUUrqCs-imMnBrWqerU,1161
 bencher/bench_cfg.py,sha256=EMoANsz0bj4lioOCO_hlJNHH9-ciK042YAXvj2pk_dE,18244
 bencher/bench_plot_server.py,sha256=HnW6XpmARweMCd-koqzu1lxuj0gA4_fP-72D3Yfy-0M,4087
 bencher/bench_report.py,sha256=b9jLksrXzBhSmvI7KetO1i0hsHknE7qk4b4k0MGGtNw,10484
 bencher/bench_runner.py,sha256=TLJGn-NO4BBBZMG8W5E6u7IPyij67Yk3S4B29ONClnk,6049
-bencher/bencher.py,sha256=kxPo4NYn7OzUgM4PzBm6CVcCIjLr2Iidh_n200n5z_g,28593
+bencher/bencher.py,sha256=ACP0OjkYi4vGO3lfBTzV5RAMqci_-ErNuEyIlnO_48o,28609
 bencher/caching.py,sha256=AusaNrzGGlj5m6zcwcqnTn55Mam2mQdF--oqelO806M,1627
 bencher/job.py,sha256=Q2zpia95Ibukk8EeFq5IBbpo2PMRe7o5keelJCJlGnI,5927
 bencher/optuna_conversions.py,sha256=XwRUQjFZs_zCZo0iT4cudstorSSXThLKOv4o75IxBrw,5427
-bencher/utils.py,sha256=vbk90hujMm11YwObreWthLqdFzydRP94ijoeT4ASAMw,3801
+bencher/utils.py,sha256=ddhTAXp9ciBhZBK5gLAJgpE6rUrMpnt4g7Wug1sSc_Q,4291
+bencher/video_writer.py,sha256=PlYLm6IBYG5MrK1J6U0ti2rvBzeMW2iXE4Pge7RsVII,523
 bencher/worker_job.py,sha256=FREi0yWQACFmH86R1j-LH72tALEFkKhLDmmoGQY9Jh4,1571
 bencher/example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bencher/example/benchmark_data.py,sha256=D9yUg_KKtqqEkAiLceodDwsv6sh7xEFWZNp6P6Y3pj4,6989
@@ -22,7 +23,7 @@ bencher/example/example_floats2D.py,sha256=D0kljoUCinMKCEW-Zg-cQ8sYu_yPCZqzKJ9tR
 bencher/example/example_holosweep.py,sha256=d9g5aWCAUb7FMZahb4t3xNs344EshPhA-8-K6K1SBXg,3180
 bencher/example/example_holosweep_objects.py,sha256=vHuAtkM1VrJelHOazn_SJfzxNywKyaMzN-DE8W7Ricc,3228
 bencher/example/example_holosweep_tap.py,sha256=3ayQ0bTj_XWP_92ifQJAhe1whwPAj_xWHPkzC7fvqAY,4540
-bencher/example/example_image.py,sha256=d9ldZDxbOXkkZGppbG9orn8sSTNzNQoKrfj_5F5TMAg,2714
+bencher/example/example_image.py,sha256=DBB7oe-67XoxPBPhUJNSr45U8t8xJvi9KiC3ptZkAHE,2713
 bencher/example/example_levels.py,sha256=rpSNB571yfMnT7iO66Ds-DPGHWzOTM9FLMNfSetJdHY,6896
 bencher/example/example_pareto.py,sha256=yyAg8Vb-5sgsS6LkYKT7T5Evcfg69FlCqCakUippSmU,2687
 bencher/example/example_sample_cache.py,sha256=7gf1BJ63VAgdqNuNXkbL9-jeTeC3kXA_PY9yG3ulTz0,4200
@@ -33,7 +34,7 @@ bencher/example/example_simple_cat.py,sha256=YFccE84g37U2M3ByWYIcGNLXWdW_ktJbbZv
 bencher/example/example_simple_float.py,sha256=X4vsH0F4fZAoO0EKB1xIzFMY0f0Wyk8LV2plPlSEsbI,1323
 bencher/example/example_strings.py,sha256=BdsEZgLT9mOxLkBKNHz2XpPwwe4SzNTdjnY1WVlOmNM,1570
 bencher/example/example_time_event.py,sha256=y1vpK7UDrObEu0Z1x3e4OQzvGCQ7pF5GZvpKLegMbYk,2158
-bencher/example/example_video.py,sha256=ZvmaJUnIw-0v8nInbUrpv6qtp1HXEGLHrGiffPwzGWU,3389
+bencher/example/example_video.py,sha256=uI8tjF8BTTD9NsrBhK5rluBujLAaOaRAB5uXAvM1fZU,3387
 bencher/example/example_workflow.py,sha256=00QnUuViMfX_PqzqkXmg1wPX6yAq7IS7mCL_RFKwrMM,6806
 bencher/example/experimental/example_bokeh_plotly.py,sha256=3jUKh8eKIAlpklKnp8UopIHhUDw1A0_5CwjeyTzbi7o,846
 bencher/example/experimental/example_hover_ex.py,sha256=qszw4FkIfqQkVviPSpmUoFOoi6PGotGbsc7Ojyx8EtU,1052
@@ -59,15 +60,15 @@ bencher/results/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bencher/results/bench_result.py,sha256=O0RxxUoudT43zukIKEmwaj8Q156UrEfwyj8NGFFwI0c,2797
 bencher/results/bench_result_base.py,sha256=_TQk3WRORY6obm57ZhsDwk6Gbq-d8u_q69OgGKtS2sU,13960
 bencher/results/float_formatter.py,sha256=sX6HNCyaXdHDxC8ybVUHwCJ3qOKbPUkBOplVIHtKWjM,1746
-bencher/results/holoview_result.py,sha256=h-Q48m2dkHLPmvHnxmREpzt7MzQ-6iYLWrURDdLPjOw,21310
+bencher/results/holoview_result.py,sha256=kcqetUFlz5HdBvV3ZjG6F-QCZabpFSNupAGYPkIZKuA,21269
 bencher/results/optuna_result.py,sha256=704l1eFJUQGTmnTaj2pIJ9ocRehLgu4PwMYmJU2rY9w,12399
-bencher/results/panel_result.py,sha256=Fkp7eWt7lzmnM5R6MdEsBeyGg7CEJmlSEHWzNujcIUc,3479
+bencher/results/panel_result.py,sha256=5oMY0KDRoGK_3LoUJB-pRy9b0QMowVvu7hR6f2-0P8c,3647
 bencher/results/plotly_result.py,sha256=vny-HPyWIWzsLWQaOvMswtgVDub9bgQzqcartlFCqjI,2368
-bencher/variables/inputs.py,sha256=5yN45tet1cPyok6R_XD8xn1FZB5OUUp_TWWF2wIVVdg,6247
-bencher/variables/parametrised_sweep.py,sha256=q633o5zFY7P0ZeHX4G3AW3jYBfwoqLZZoUeibcln1u0,7483
+bencher/variables/inputs.py,sha256=HYWEK75qfzk86dWvZbb6TeZr6yEvXgXILdddAqGuvEs,6392
+bencher/variables/parametrised_sweep.py,sha256=96fLNjq1x7tGSwGCVshR3_LK9PRHhLipj8pbJ5umbnA,7029
 bencher/variables/results.py,sha256=yXdLjfGPSerWhipw_GDZJpBB8KQgCqcdreb8WjIeah8,5119
 bencher/variables/sweep_base.py,sha256=I1LEeG1y5Jsw0a-Ik03t0tSzcfENht2GmBECJ3KNs28,6559
 bencher/variables/time.py,sha256=Le7s8_oUYJD4wCqwQw-a_FRDpYQOi8CqMbGYsBF07jg,2860
-holobench-1.3.1.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
-holobench-1.3.1.dist-info/METADATA,sha256=vSgr3cL-IZvlxj-D9jz245_78Lj85EPycjjP7ceaALI,4956
-holobench-1.3.1.dist-info/RECORD,,
+holobench-1.3.2.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
+holobench-1.3.2.dist-info/METADATA,sha256=rDXDdz5mTtMkOgBnmiVjJfU_JPIgZxDu1VnGlh_KTQw,4994
+holobench-1.3.2.dist-info/RECORD,,