From 95d0a1bcfe9fcfbc8accc08699d31054951082d4 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Mon, 24 Nov 2025 00:12:01 +0100 Subject: [PATCH] refactor(ReforceXY): remove unnecessary import MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- ReforceXY/user_data/freqaimodels/ReforceXY.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py index 3f22e9c..6b14f0f 100644 --- 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) -- 2.43.0