zhuwq0 commited on
Commit
4da07fe
·
1 Parent(s): c8f6682

add default values

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -56,6 +56,8 @@ def predict(mseeds=[], waveforms="", stations=""):
56
  waveforms = np.array(waveforms)
57
  print(f"{waveforms.shape = }")
58
  picks = []
 
 
59
  for mseed in mseeds:
60
  file = mseed.name # if not isinstance(input, str) else input
61
  mseed = obspy.read(file)
@@ -83,7 +85,7 @@ def predict(mseeds=[], waveforms="", stations=""):
83
 
84
  picks.extend(tmp_picks)
85
 
86
- if len(picks) > 0
87
  picks_df = pd.DataFrame(picks)
88
  picks_df.to_csv("picks.csv", index=False)
89
  else:
 
56
  waveforms = np.array(waveforms)
57
  print(f"{waveforms.shape = }")
58
  picks = []
59
+ if mseeds is None:
60
+ mseeds = []
61
  for mseed in mseeds:
62
  file = mseed.name # if not isinstance(input, str) else input
63
  mseed = obspy.read(file)
 
85
 
86
  picks.extend(tmp_picks)
87
 
88
+ if len(picks) > 0:
89
  picks_df = pd.DataFrame(picks)
90
  picks_df.to_csv("picks.csv", index=False)
91
  else: