adding error checking

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())
 

rowers/tasks.py

@@ -2802,7 +2802,7 @@ def handle_update_wps(rid,types,ids,mode,debug=False,**kwargs):
     df = getsmallrowdata_db(['time','driveenergy'],ids=ids)
     try:
         mask = df['driveenergy'] > 100
-    except KeyError:
+    except (KeyError, TypeError):
         return 0
     wps_median = int(df.loc[mask,'driveenergy'].median())