hot fix: if_overridden_by_trainer

AllenNeuralDynamics · Apr 9, 2024 · cb2f4b3 · cb2f4b3
1 parent 9ec86e6
commit cb2f4b3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/code/Home.py b/code/Home.py
@@ -662,6 +662,7 @@ def _get_data_source(rig):
                      'current_stage_actual': 'None',
                      'has_video': False,
                      'has_ephys': False,
+                     'if_overriden_by_trainer': False,
                      }
     _df.fillna(filled_values, inplace=True)
 
@@ -685,6 +686,9 @@ def _get_data_source(rig):
     # drop 'bpod_backup_' columns
     _df.drop([col for col in _df.columns if 'bpod_backup_' in col], axis=1, inplace=True)
 
+    # fix if_overriden_by_trainer
+    _df['if_overriden_by_trainer'] = _df['if_overriden_by_trainer'].astype(bool)
+
     # _df = _df.merge(
     #     diff_relative_weight_next_day, how='left', on=['h2o', 'session'])