]> Piment Noir Git Repositories - freqai-strategies.git/commitdiff
refactor(reforcexy): refine typing
authorJérôme Benoit <jerome.benoit@piment-noir.org>
Thu, 25 Sep 2025 19:49:48 +0000 (21:49 +0200)
committerJérôme Benoit <jerome.benoit@piment-noir.org>
Thu, 25 Sep 2025 19:49:48 +0000 (21:49 +0200)
Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>
ReforceXY/user_data/freqaimodels/ReforceXY.py

index 143a301c987ceead68e6450edd83d7dbccc37b4d..1e1a471ef362709d75bdca28026399e2dbf597dd 100644 (file)
@@ -53,6 +53,7 @@ from stable_baselines3.common.utils import ConstantSchedule, set_random_seed
 from stable_baselines3.common.vec_env import (
     DummyVecEnv,
     SubprocVecEnv,
+    VecEnv,
     VecFrameStack,
     VecMonitor,
 )
@@ -898,7 +899,7 @@ class ReforceXY(BaseReinforcementLearningModel):
         seed: Optional[int] = None,
         env_info: Optional[Dict[str, Any]] = None,
         trial: Optional[Trial] = None,
-    ) -> Tuple[BaseEnvironment, BaseEnvironment]:
+    ) -> Tuple[VecEnv, VecEnv]:
         if (
             train_df is None
             or test_df is None