fix

AI4EPS · Nov 7, 2024 · 39e662b · 39e662b
1 parent 2f467d1
commit 39e662b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/run_phasenet_plus.py b/scripts/run_phasenet_plus.py
@@ -68,7 +68,7 @@ def run_phasenet(
 
     # %% skip processed files
     if not overwrite:
-        processed = sorted(glob(f"{root_path}/{result_path}/picks_phasenet_plus/????/???/*.csv"))
+        processed = sorted(glob(f"{root_path}/{result_path}/picks_phasenet_plus/????/???/??/*.csv"))
         # processed = sorted(glob(f"{root_path}/{result_path}/picks_phasenet_plus/????/???/*.csv"))
         processed = ["/".join(f.replace(".csv", "").split("/")[-subdir - 1 :]) for f in processed]
         processed = [p[:-1] for p in processed]  ## remove the channel suffix