From: Jérôme Benoit <jerome.benoit@piment-noir.org>
Date: Sun, 23 Nov 2025 23:12:01 +0000 (+0100)
Subject: refactor(ReforceXY): remove unnecessary import
X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=95d0a1bcfe9fcfbc8accc08699d31054951082d4;p=freqai-strategies.git

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 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)