Added a missing end_timer

rcap107 · Sep 21, 2023 · 703ba44 · 703ba44
1 parent a6b52d5
commit 703ba44
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/methods/evaluation.py b/src/methods/evaluation.py
@@ -420,6 +420,8 @@ def run_on_full_join(
         eval_data = prepare_table_for_evaluation(merged_test)
         y_test = eval_data[target_column].cast(pl.Float64)
         eval_data = eval_data.drop(target_column).to_pandas()
+        run_logger.end_time("eval")
+        raw_logger.end_time("eval")
 
         y_pred = model.predict(eval_data)