fix(reforcexy): proper compute fit time

Signed-off-by: Jérôme Benoit <jerome.benoit@sap.com>

diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py
index c0b9aee3127f4e9ec886d59810bec82151ee4dda..bd68492e41371f5f870f5b960fe05287420dc8ba 100644 (file)
--- a/ReforceXY/user_data/freqaimodels/ReforceXY.py
+++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py
@@ -331,6 +331,7 @@ class ReforceXY(BaseReinforcementLearningModel):
         )
         logger.info("Hyperopt: %s", self.hyperopt)
 
+        start = time.time()
         if self.hyperopt:
             model_params = self.study(train_df, total_timesteps, dk)
         else:
@@ -362,6 +363,8 @@ class ReforceXY(BaseReinforcementLearningModel):
         finally:
             if self.progressbar_callback:
                 self.progressbar_callback.on_training_end()
+        time_spent = time.time() - start
+        self.dd.update_metric_tracker("fit_time", time_spent, dk.pair)
 
         if Path(dk.data_path / "best_model.zip").is_file():
             logger.info("Callback found a best model.")