Forest Model related data additions

While testing model integration, 2 forest model features specific features are added in the `TestForestModelIntegration.py` file rather than in `entry.py` file.
e-mission · Mar 21, 2024 · 35a1346 · 35a1346
1 parent 1b523ed
commit 35a1346
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/emission/core/wrapper/entry.py b/emission/core/wrapper/entry.py
@@ -182,9 +182,6 @@ def create_fake_entry(user_id, key, data, write_ts, create_id=False):
       result_entry.user_id = user_id
       result_entry.metadata = ecwm.Metadata.create_metadata_for_fake_result(key, write_ts)
       result_entry.data = data
-      #necessary values required by forest model
-      result_entry['data']['start_local_dt']=result_entry['metadata']['write_local_dt']
-      result_entry['data']['end_local_dt']=result_entry['metadata']['write_local_dt']
       result_entry._populateDependencies()
       return result_entry
 

diff --git a/emission/tests/modellingTests/TestForestModelIntegration.py b/emission/tests/modellingTests/TestForestModelIntegration.py
@@ -54,6 +54,11 @@ def setUp(self):
             threshold=0.004, # ~400m
             has_label_p=0.9
         )
+
+        for result_entry in train:
+            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(train)
         # confirm data write did not fail
         check_data = esda.get_entries(key="analysis/confirmed_trip", user_id=self.testUUID, time_query=None)

diff --git a/emission/tests/modellingTests/TestRunForestModel.py b/emission/tests/modellingTests/TestRunForestModel.py
@@ -5,8 +5,6 @@
 import emission.analysis.modelling.trip_model.model_type as eamumt
 import emission.analysis.modelling.trip_model.model_storage as eamums
 import emission.analysis.modelling.trip_model.models as eamtm
-logger=logging.getLogger("")
-logger.setLevel(logging.DEBUG)
 import emission.storage.timeseries.abstract_timeseries as esta
 import emission.tests.modellingTests.modellingTestAssets as etmm
 import emission.storage.decorations.analysis_timeseries_queries as esda