format pr dataset file

NeuroBench · May 21, 2024 · bb872c9 · bb872c9
1 parent cacf39c
commit bb872c9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/neurobench/datasets/primate_reaching.py b/neurobench/datasets/primate_reaching.py
@@ -224,7 +224,9 @@ def load_data(self):
 
         # Define the segments' start & end indices
         self.start_end_indices = np.array(self.get_flag_index(target_pos))
-        self.time_segments = np.array(self.split_into_segments(self.start_end_indices, target_pos.shape[1]))
+        self.time_segments = np.array(
+            self.split_into_segments(self.start_end_indices, target_pos.shape[1])
+        )
 
         spike_train = np.zeros((*spikes.shape, len(new_t)), dtype=np.int8)