Merge pull request #868 from SimonLafran/fix-addinfo-indexing

[FIX] KeyError in _eventrelated_addinfo
neuropsychology · Jul 17, 2023 · a802b57 · a802b57
2 parents 1c1651d + ac08ef3
commit a802b57
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neurokit2/epochs/eventrelated_utils.py b/neurokit2/epochs/eventrelated_utils.py
@@ -46,7 +46,7 @@ def _eventrelated_sanitizeinput(epochs, what="ecg", silent=False):
 def _eventrelated_addinfo(epoch, output={}):
     # Add label
     if "Index" in epoch.columns:
-        output["Event_Onset"] = epoch.loc[np.min(np.abs(epoch.index))]["Index"]
+        output["Event_Onset"] = epoch.iloc[np.argmin(np.abs(epoch.index))]["Index"]
 
     # Add label
     if "Label" in epoch.columns and len(set(epoch["Label"])) == 1: