Fixed function call to predict_labels_with_n in TestRunGreedyModel

Updated function call to predict_labels_with_n() as per latest changes to include trip_list and user_id Fixed typo in eacilp for trip.get_id()
e-mission · Nov 17, 2023 · 30c02a4 · 30c02a4
1 parent fb278ff
commit 30c02a4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/emission/analysis/classification/inference/labels/pipeline.py b/emission/analysis/classification/inference/labels/pipeline.py
@@ -106,7 +106,7 @@ def compute_and_save_ensemble(self, trip_list, predictions_dict):
         il_list = []
         for trip, key in zip(trip_list, predictions_dict):
             # Print trip ids to see if order maintained: 
-            print(f"Trip_ID: {trip.get_id} and Pred_Key: {key}")
+            print(f"Trip_ID: {trip.get_id()} and Pred_Key: {key}")
             il = ecwl.Labelprediction()
             il.trip_id = trip.get_id()
             il.start_ts = trip["data"]["start_ts"]

diff --git a/emission/tests/modellingTests/TestRunGreedyModel.py b/emission/tests/modellingTests/TestRunGreedyModel.py
@@ -164,7 +164,8 @@ def test1RoundTripGreedySimilarityBinning(self):
             destination=self.destination
         )
         prediction, n = eamur.predict_labels_with_n(
-            trip = test,
+            user_id = self.user_id,
+            trip = [test],
             model_type=eamumt.ModelType.GREEDY_SIMILARITY_BINNING,
             model_storage=eamums.ModelStorage.DOCUMENT_DATABASE,
             model_config=greedy_model_config