require stim offset to define epoch (#529)

Co-authored-by: Tom Chartrand <[email protected]>
AllenInstitute · Aug 23, 2021 · dc4ade5 · dc4ade5
1 parent 1129819
commit dc4ade5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ipfx/epochs.py b/ipfx/epochs.py
@@ -113,7 +113,7 @@ def get_stim_epoch(i, test_pulse=True):
     if test_pulse:
         di_idx = di_idx[2:]     # drop the first up/down (test pulse) if present
 
-    if len(di_idx) == 0:    # if no stimulus is found
+    if len(di_idx) < 2:    # if no stimulus is found
         return None
 
     start_idx = di_idx[0] + 1   # shift by one to compensate for diff()