adding error checking

2021-05-24 21:58:31 +02:00
parent 5437ff6e2f
commit 37a063950f
3 changed files with 2 additions and 2 deletions
--- a/rowers/dataprep.py
+++ b/rowers/dataprep.py
@@ -1286,7 +1286,7 @@ def update_wps(r,types,mode='water',asynchron=True):

    try:
        mask = df['driveenergy'] > 100
-    except KeyError:
+    except (KeyError, TypeError):
        return False
    wps_median = int(df.loc[mask,'driveenergy'].median())