Reduce conversions between lists and arrays

LorenFrankLab · Oct 2, 2023 · 6e000d1 · 6e000d1
1 parent df2ed88
commit 6e000d1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/spikegadgets_to_nwb/convert_ephys.py b/src/spikegadgets_to_nwb/convert_ephys.py
@@ -146,7 +146,7 @@ def _get_data(self, selection: Tuple[slice]) -> np.ndarray:
             # find the stream where this piece of slice begins
             io_stream = np.argmin(i >= file_start_ind) - 1
             # get the data from that stream
-            data.extend(
+            data.append(
                 (
                     self.neo_io[io_stream].get_analogsignal_chunk(
                         block_index=self.block_index,
@@ -170,7 +170,7 @@ def _get_data(self, selection: Tuple[slice]) -> np.ndarray:
                 time_index[-1] - i,  # if finished in this stream
             )
 
-        data = (np.array(data) * self.conversion).astype("int16")
+        data = (np.concatenate(data) * self.conversion).astype("int16")
         return data
 
     def _get_maxshape(self) -> Tuple[int, int]: