Remove prints and fix MEArec get_traces

SpikeInterface · Dec 8, 2020 · 2946f9b · 2946f9b
1 parent 2d00235
commit 2946f9b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/spikeextractors/extractors/mearecextractors/mearecextractors.py b/spikeextractors/extractors/mearecextractors/mearecextractors.py
@@ -80,11 +80,11 @@ def get_traces(self, channel_ids=None, start_frame=None, end_frame=None):
             sorted_channel_ids = np.sort(channel_ids)
             sorted_idx = np.array([list(sorted_channel_ids).index(ch) for ch in channel_ids])
             recordings = self._recordings[start_frame:end_frame, sorted_channel_ids]
-            return np.array(recordings[sorted_idx]).transpose()
+            return np.array(recordings[:, sorted_idx]).T
         else:
             if sorted(channel_ids) == channel_ids and np.all(np.diff(channel_ids) == 1):
                 channel_ids = slice(channel_ids[0], channel_ids[0] + len(channel_ids))
-            return np.array(self._recordings[start_frame:end_frame, channel_ids]).transpose()
+            return np.array(self._recordings[start_frame:end_frame, channel_ids]).T
 
     @staticmethod
     def write_recording(recording, save_path, check_suffix=True):

diff --git a/spikeextractors/extractors/nwbextractors/nwbextractors.py b/spikeextractors/extractors/nwbextractors/nwbextractors.py
@@ -289,7 +289,6 @@ def get_traces(self, channel_ids=None, start_frame=None, end_frame=None):
                 # to be indexed out of order
                 sorted_channel_ids = np.sort(channel_ids)
                 sorted_idx = np.array([list(sorted_channel_ids).index(ch) for ch in channel_ids])
-                print(sorted_idx)
                 recordings = es.data[start_frame:end_frame, sorted_channel_ids].T
                 traces = recordings[sorted_idx, :]
             else: