return empty when missing thetacycles

ryanharvey1 · Dec 20, 2024 · dc40e81 · dc40e81
1 parent b299afd
commit dc40e81
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/neuro_py/io/loading.py b/neuro_py/io/loading.py
@@ -954,6 +954,12 @@ def load_theta_cycles(
     filename = os.path.join(
         basepath, os.path.basename(basepath) + ".thetacycles.events.mat"
     )
+    if not os.path.exists(filename):
+        warnings.warn("file does not exist")
+        if return_epoch_array:
+            return nel.EpochArray()
+        return pd.DataFrame()
+
     data = sio.loadmat(filename, simplify_cells=True)
     df = pd.DataFrame()
     df["start"] = data["thetacycles"]["timestamps"][:, 0]