diff --git a/code/Home.py b/code/Home.py index 39cfd2e..61a3f76 100644 --- a/code/Home.py +++ b/code/Home.py @@ -663,7 +663,12 @@ def _get_data_source(rig): _df.fillna(filled_values, inplace=True) # Remove abnormal values - _df.loc[_df['weight_after'] > 100, ['weight_after', 'weight_after_ratio']] = np.nan + _df.loc[_df['weight_after'] > 100, + ['weight_after', 'weight_after_ratio', 'water_in_session_total', 'water_after_session', 'water_day_total'] + ] = np.nan + + _df.loc[_df['water_in_session_manual'] > 100, + ['water_in_session_manual', 'water_in_session_total', 'water_after_session']] = np.nan # foraging performance = foraing_eff * finished_rate if 'foraging_performance' not in _df.columns: