Update data_pipeline.py

YijianZhou · Apr 8, 2021 · 6722e8c · 6722e8c
1 parent c17ad74
commit 6722e8c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/data_pipeline.py b/data_pipeline.py
@@ -65,7 +65,8 @@ def get_picks(date, pick_dir):
              ('p_snr','O'),
              ('freq_dom','O')]
     fname = str(date.date) + '.pick'
-    pick_path = os.path.join(pick_dir, fname)
+    pick_path = os.path.join(pick_dir, fname)、
+    if not os.path.exists(pick_path): return np.array([], dtype=dtype)
     f=open(pick_path); lines=f.readlines(); f.close()
     for line in lines:
         codes = line.split(',')