test memmap fix (#54)

zm711 · Aug 8, 2024 · df72e31 · df72e31
1 parent df06e9c
commit df72e31
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/spikeanalysis/stimulus_data.py b/src/spikeanalysis/stimulus_data.py
@@ -578,7 +578,12 @@ def _intan_neo_read_no_dig(self, reader: neo.rawio.IntanRawIO, time_slice: tuple
             the raw digital data stored. Cannot be used. Must be processed first.
 
         """
-        digital_memmap = reader._raw_data["DIGITAL-IN"]  # directly grab memory map from neo
+        try:
+            digital_memmap = reader._raw_data['USB board digital input channel'] # this will be the field name now
+        except ValueError:
+            # As of PR1491 the name has changed keep this for back compatibility
+            digital_memmap = reader._raw_data["DIGITAL-IN"]  # directly grab memory map from neo
+
         dig_size = digital_memmap.size
         dig_shape = digital_memmap.shape
         # below we have all the shaping information necessary