Merge pull request #2828 from alejoe91/fix-zarr-times

Zarr: extract time vector once and for all!
SpikeInterface · May 10, 2024 · cd55f91 · cd55f91
2 parents 551a3b4 + 37a01d3
commit cd55f91
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/spikeinterface/core/zarrextractors.py b/src/spikeinterface/core/zarrextractors.py
@@ -72,7 +72,7 @@ def __init__(self, folder_path: Path | str, storage_options: dict | None = None)
             time_kwargs = {}
             time_vector = self._root.get(f"times_seg{segment_index}", None)
             if time_vector is not None:
-                time_kwargs["time_vector"] = time_vector
+                time_kwargs["time_vector"] = time_vector[:]
             else:
                 if t_starts is None:
                     t_start = None

diff --git a/src/spikeinterface/generation/template_database.py b/src/spikeinterface/generation/template_database.py
@@ -2,7 +2,7 @@
 import zarr
 
 
-def fetch_templates_from_database(dataset="test_templates") -> Templates:
+def fetch_templates_from_database(dataset="test_templates.zarr") -> Templates:
 
     s3_path = f"s3://spikeinterface-template-database/{dataset}/"
     zarr_group = zarr.open_consolidated(s3_path, storage_options={"anon": True})