avoid creating many Annotations objects in for loop

mne-tools · Oct 9, 2022 · ff0aad8 · ff0aad8
1 parent 79aeaa4
commit ff0aad8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mne_bids/read.py b/mne_bids/read.py
@@ -491,8 +491,8 @@ def _handle_events_reading(events_fname, raw):
                                         description=descriptions)
     raw.set_annotations(annot_from_events)
 
-    annot_idx_to_keep = [idx for idx, annot in enumerate(annot_from_raw)
-                         if annot['description'] in ANNOTATIONS_TO_KEEP]
+    annot_idx_to_keep = [idx for idx, descr in enumerate(annot_from_raw.description)
+                         if descr in ANNOTATIONS_TO_KEEP]
     annot_to_keep = annot_from_raw[annot_idx_to_keep]
 
     if len(annot_to_keep):