prevents crashing on short trials

Author: neuromusic

visual_behavior/validation/extended_trials.py

@@ -168,21 +168,24 @@ def count_stimuli_per_trial(trials, visual_stimuli):
         elif all(pd.isnull(visual_stimuli.orientation)) == False:
             col_to_check = 'orientation'
 
-        # get the index of the first stimulus that started on or before the trial start
-        start_stim = visual_stimuli[
-            (visual_stimuli.frame <= trial.startframe + 1) &
-            (visual_stimuli.end_frame >= trial.startframe + 1)
-        ].index[-1]
-
-        # get the index of the last stimulus that ended on or before the trial end
-        end_stim = visual_stimuli[
-            (visual_stimuli.frame <= trial.endframe)
-        ].index[-1]
-
-        # get all unique stimuli on this trial
-        stimuli = np.unique(visual_stimuli.loc[start_stim:end_stim][col_to_check])
-        # add to array
-        stimuli_per_trial[idx] = len(stimuli)
+        try:
+            # get the index of the first stimulus that started on or before the trial start
+            start_stim = visual_stimuli[
+                (visual_stimuli.frame <= trial.startframe + 1) &
+                (visual_stimuli.end_frame >= trial.startframe + 1)
+            ].index[-1]
+
+            # get the index of the last stimulus that ended on or before the trial end
+            end_stim = visual_stimuli[
+                (visual_stimuli.frame <= trial.endframe)
+            ].index[-1]
+
+            # get all unique stimuli on this trial
+            stimuli = np.unique(visual_stimuli.loc[start_stim:end_stim][col_to_check])
+            # add to array
+            stimuli_per_trial[idx] = len(stimuli)
+        except IndexError:
+            stimuli_per_trial[idx] = 0
     return stimuli_per_trial