Bugfix: savecsv no longer raises attributeerror, properly includes co…

…dec json and zarr version
royerlab · Jul 18, 2024 · 2442983 · 2442983
1 parent 0a0bee0
commit 2442983
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "czpeedy"
-version = "0.2.0"
+version = "0.2.1"
 description = "A command-line tool used to determine the tensorstore settings which yield the fastest write speed on a given machine."
 authors = [
     { name = "Seth Hinz", email = "[email protected]" }

diff --git a/src/czpeedy/runner.py b/src/czpeedy/runner.py
@@ -119,6 +119,7 @@ def save_results_csv(self, path: Path):
                     "mean write time (s)",
                     "write std.dev (s)",
                     "dtype",
+                    "zarr version",
                     "clevel",
                     "compressor",
                     "shuffle",
@@ -133,7 +134,8 @@ def save_results_csv(self, path: Path):
                     [
                         f"{np.mean(timings):.2f}",
                         f"{np.std(timings):.2f}",
-                        trial_param.dtype_json(),
+                        trial_param.dtype_json_v2() if trial_param.zarr_version == 2 else trial_param.dtype_json_v3(),
+                        trial_param.zarr_version,
                         trial_param.clevel,
                         trial_param.compressor,
                         trial_param.shuffle,