From: Jérôme Benoit <jerome.benoit@piment-noir.org>
Date: Thu, 25 Sep 2025 19:49:48 +0000 (+0200)
Subject: refactor(reforcexy): refine typing
X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=5d4f49553e90d6fb9b19795bf2450a464d83cf38;p=freqai-strategies.git

refactor(reforcexy): refine typing

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 143a301..1e1a471 100644
--- a/ReforceXY/user_data/freqaimodels/ReforceXY.py
+++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py
@@ -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