more minor

AllenNeuralDynamics · Apr 8, 2024 · 2b3f61c · 2b3f61c
1 parent d57f7a5
commit 2b3f61c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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: