FIX: early return in find_breaks if no annotations in raw

lina-usc · Nov 13, 2024 · dbeea81 · dbeea81
1 parent 68c1784
commit dbeea81
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/pylossless/pipeline.py b/pylossless/pipeline.py
@@ -718,6 +718,9 @@ def find_breaks(self):
         """
         if "find_breaks" not in self.config or not self.config["find_breaks"]:
             return
+        if not self.raw.annotations:
+            logger.debug("No annotations found in raw object. Skipping find_breaks.")
+            return
         breaks = annotate_break(self.raw, **self.config["find_breaks"])
         self.raw.set_annotations(breaks + self.raw.annotations)
 

diff --git a/pylossless/tests/test_pipeline.py b/pylossless/tests/test_pipeline.py
@@ -43,6 +43,9 @@ def test_find_breaks(logging):
         pipeline.find_breaks(message="Looking for break periods between tasks")
     else:
         pipeline.find_breaks()
+        # Now explicitly remove annotations and make sure we avoid MNE's error.
+        pipeline.raw.set_annotations(None)
+        pipeline.find_breaks()
     Path(config_fname).unlink()  # delete config file