FIX: early return in find_breaks if no annotations in raw

Author: scott-huberty

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)
 

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