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 143a301c987ceead68e6450edd83d7dbccc37b4d..1e1a471ef362709d75bdca28026399e2dbf597dd 100644 (file)
--- 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