refactor(ReforceXY): remove unnecessary import

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>

diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py
index 3f22e9c929586e353eb92583144b45d5e9e89345..6b14f0fb905a3f407856ffc4d8cf06df18cce945 100644 (file)
--- a/ReforceXY/user_data/freqaimodels/ReforceXY.py
+++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py
@@ -52,7 +52,6 @@ from optuna.storages import (
 )
 from optuna.storages.journal import JournalFileBackend
 from optuna.study import Study, StudyDirection
-from optuna.study.study import ObjectiveFuncType
 from pandas import DataFrame, merge
 from sb3_contrib.common.maskable.callbacks import MaskableEvalCallback
 from sb3_contrib.common.maskable.utils import is_masking_supported
@@ -1160,11 +1159,8 @@ class ReforceXY(BaseReinforcementLearningModel):
         hyperopt_failed = False
         start_time = time.time()
         try:
-            objective: ObjectiveFuncType = lambda trial: self.objective(
-                trial, dk, total_timesteps
-            )
             study.optimize(
-                objective,
+                lambda trial: self.objective(trial, dk, total_timesteps),
                 n_trials=self.optuna_n_trials,
                 timeout=(
                     hours_to_seconds(self.optuna_timeout_hours)