add default values
Browse files
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:
|