fix: update spike interface version (#19)

AllenNeuralDynamics · May 5, 2024 · 05d44ce · 05d44ce
1 parent ad4d744
commit 05d44ce
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -23,7 +23,7 @@ dependencies = [
 
 [project.optional-dependencies]
 ephys = [
-    'spikeinterface[full]>=0.100.0',
+    'spikeinterface[full]>=0.100.5',
     'probeinterface==0.2.21',
     'wavpack-numcodecs>=0.1.3,<=0.1.5',
     'zarr==2.16.1',

diff --git a/src/aind_data_transformation/ephys/ephys_job.py b/src/aind_data_transformation/ephys/ephys_job.py
@@ -358,10 +358,12 @@ def _compress_and_write_block(
                     f" and might lead to errors. Use a shorter destination "
                     f"path."
                 )
+            # compression for times is disabled
             _ = rec.save(
                 format=output_format,
                 folder=zarr_path,
                 compressor=compressor,
+                compressor_by_dataset=dict(times=None),
                 **job_kwargs,
             )
 

diff --git a/tests/test_ephys_job.py b/tests/test_ephys_job.py
@@ -537,6 +537,7 @@ def test_compress_and_write_scaled_blocks(
                         num_encoding_threads=1,
                         shaping_weight=0.0,
                     ),
+                    compressor_by_dataset={"times": None},
                     n_jobs=1,
                 ),
                 call(
@@ -557,6 +558,7 @@ def test_compress_and_write_scaled_blocks(
                         num_encoding_threads=1,
                         shaping_weight=0.0,
                     ),
+                    compressor_by_dataset={"times": None},
                     n_jobs=1,
                 ),
                 call(
@@ -577,6 +579,7 @@ def test_compress_and_write_scaled_blocks(
                         num_encoding_threads=1,
                         shaping_weight=0.0,
                     ),
+                    compressor_by_dataset={"times": None},
                     n_jobs=1,
                 ),
             ]
@@ -601,6 +604,7 @@ def test_compress_and_write_scaled_blocks(
                         num_encoding_threads=1,
                         shaping_weight=0.0,
                     ),
+                    compressor_by_dataset={"times": None},
                     n_jobs=1,
                 ),
                 call(
@@ -621,6 +625,7 @@ def test_compress_and_write_scaled_blocks(
                         num_encoding_threads=1,
                         shaping_weight=0.0,
                     ),
+                    compressor_by_dataset={"times": None},
                     n_jobs=1,
                 ),
                 call(
@@ -641,6 +646,7 @@ def test_compress_and_write_scaled_blocks(
                         num_encoding_threads=1,
                         shaping_weight=0.0,
                     ),
+                    compressor_by_dataset={"times": None},
                     n_jobs=1,
                 ),
                 call(
@@ -661,6 +667,7 @@ def test_compress_and_write_scaled_blocks(
                         num_encoding_threads=1,
                         shaping_weight=0.0,
                     ),
+                    compressor_by_dataset={"times": None},
                     n_jobs=1,
                 ),
                 call(
@@ -681,6 +688,7 @@ def test_compress_and_write_scaled_blocks(
                         num_encoding_threads=1,
                         shaping_weight=0.0,
                     ),
+                    compressor_by_dataset={"times": None},
                     n_jobs=1,
                 ),
                 call(
@@ -701,6 +709,7 @@ def test_compress_and_write_scaled_blocks(
                         num_encoding_threads=1,
                         shaping_weight=0.0,
                     ),
+                    compressor_by_dataset={"times": None},
                     n_jobs=1,
                 ),
             ]
@@ -757,6 +766,7 @@ def test_compress_and_write_read_blocks(
                         num_encoding_threads=1,
                         shaping_weight=0.0,
                     ),
+                    compressor_by_dataset={"times": None},
                     n_jobs=1,
                 ),
                 call(
@@ -777,6 +787,7 @@ def test_compress_and_write_read_blocks(
                         num_encoding_threads=1,
                         shaping_weight=0.0,
                     ),
+                    compressor_by_dataset={"times": None},
                     n_jobs=1,
                 ),
                 call(
@@ -797,6 +808,7 @@ def test_compress_and_write_read_blocks(
                         num_encoding_threads=1,
                         shaping_weight=0.0,
                     ),
+                    compressor_by_dataset={"times": None},
                     n_jobs=1,
                 ),
                 call(
@@ -817,6 +829,7 @@ def test_compress_and_write_read_blocks(
                         num_encoding_threads=1,
                         shaping_weight=0.0,
                     ),
+                    compressor_by_dataset={"times": None},
                     n_jobs=1,
                 ),
                 call(
@@ -837,6 +850,7 @@ def test_compress_and_write_read_blocks(
                         num_encoding_threads=1,
                         shaping_weight=0.0,
                     ),
+                    compressor_by_dataset={"times": None},
                     n_jobs=1,
                 ),
                 call(
@@ -857,6 +871,7 @@ def test_compress_and_write_read_blocks(
                         num_encoding_threads=1,
                         shaping_weight=0.0,
                     ),
+                    compressor_by_dataset={"times": None},
                     n_jobs=1,
                 ),
                 call(
@@ -877,6 +892,7 @@ def test_compress_and_write_read_blocks(
                         num_encoding_threads=1,
                         shaping_weight=0.0,
                     ),
+                    compressor_by_dataset={"times": None},
                     n_jobs=1,
                 ),
                 call(
@@ -897,6 +913,7 @@ def test_compress_and_write_read_blocks(
                         num_encoding_threads=1,
                         shaping_weight=0.0,
                     ),
+                    compressor_by_dataset={"times": None},
                     n_jobs=1,
                 ),
                 call(
@@ -917,6 +934,7 @@ def test_compress_and_write_read_blocks(
                         num_encoding_threads=1,
                         shaping_weight=0.0,
                     ),
+                    compressor_by_dataset={"times": None},
                     n_jobs=1,
                 ),
             ],