Merge pull request #3391 from alejoe91/reset-times

Reset times 3: fix case when no time vector

Author: samuelgarcia

src/spikeinterface/core/baserecording.py

@@ -503,8 +503,8 @@ def reset_times(self):
         segment's sampling frequency is set to the recording's sampling frequency.
         """
         for segment_index in range(self.get_num_segments()):
+            rs = self._recording_segments[segment_index]
             if self.has_time_vector(segment_index):
-                rs = self._recording_segments[segment_index]
                 rs.time_vector = None
             rs.t_start = None
             rs.sampling_frequency = self.sampling_frequency

src/spikeinterface/core/tests/test_baserecording.py

@@ -298,6 +298,9 @@ def test_BaseRecording(create_cache_folder):
         assert time_info["time_vector"] is None
         assert time_info["sampling_frequency"] == rec.sampling_frequency
 
+    # resetting time again should be ok
+    rec.reset_times()
+
     # test 3d probe
     rec_3d = generate_recording(ndim=3, num_channels=30)
     locations_3d = rec_3d.get_property("location")