From 2b3f61ceed14bdb4d6d99a4416895ad0427727da Mon Sep 17 00:00:00 2001
From: hh_itx_win10 <houhan@gmail.com>
Date: Sun, 7 Apr 2024 19:46:35 -0700
Subject: [PATCH] more minor

---
 code/Home.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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: