diff --git a/rowers/dataprep.py b/rowers/dataprep.py index 99fafaaf..a04a11ee 100644 --- a/rowers/dataprep.py +++ b/rowers/dataprep.py @@ -1766,6 +1766,7 @@ def prepmultipledata(ids, verbose=False): def read_cols_df_sql(ids, columns, convertnewtons=True): # drop columns that are not in offical list # axx = [ax[0] for ax in axes] + prepmultipledata(ids) axx = [f.name for f in StrokeData._meta.get_fields()] extracols = [] @@ -1802,9 +1803,11 @@ def read_cols_df_sql(ids, columns, convertnewtons=True): ids=tuple(ids), )) + connection = engine.raw_connection() df = pd.read_sql_query(query, engine) + df = df.fillna(value=0) if 'peakforce' in columns: