diff --git a/emission/tests/modellingTests/TestForestModelLoadandSave.py b/emission/tests/modellingTests/TestForestModelLoadandSave.py index 37768a689..38610fad4 100644 --- a/emission/tests/modellingTests/TestForestModelLoadandSave.py +++ b/emission/tests/modellingTests/TestForestModelLoadandSave.py @@ -67,6 +67,10 @@ def setUp(self): has_label_p=self.has_label_percent ) + for result_entry in test_data: + result_entry['data']['start_local_dt']=result_entry['metadata']['write_local_dt'] + result_entry['data']['end_local_dt']=result_entry['metadata']['write_local_dt'] + ts.bulk_insert(test_data) # confirm data write did not fail