diff --git a/stable_baselines3/common/atari_wrappers.py b/stable_baselines3/common/atari_wrappers.py index a28606498..bf89102da 100644 --- a/stable_baselines3/common/atari_wrappers.py +++ b/stable_baselines3/common/atari_wrappers.py @@ -11,7 +11,7 @@ except ImportError: cv2 = None -from stable_baselines3.common.type_aliases import gymnasium as gym26ResetReturn, Gym26StepReturn +from stable_baselines3.common.type_aliases import Gym26ResetReturn, Gym26StepReturn class NoopResetEnv(gym.Wrapper): diff --git a/stable_baselines3/common/envs/bit_flipping_env.py b/stable_baselines3/common/envs/bit_flipping_env.py index 50e655a9e..ea9d723da 100644 --- a/stable_baselines3/common/envs/bit_flipping_env.py +++ b/stable_baselines3/common/envs/bit_flipping_env.py @@ -5,7 +5,7 @@ from gymnasium import Env, spaces from gym.envs.registration import EnvSpec -from stable_baselines3.common.type_aliases import gymnasium as gym26StepReturn +from stable_baselines3.common.type_aliases import Gym26StepReturn class BitFlippingEnv(Env): diff --git a/stable_baselines3/common/envs/identity_env.py b/stable_baselines3/common/envs/identity_env.py index 482de89fe..03fc1dbd2 100644 --- a/stable_baselines3/common/envs/identity_env.py +++ b/stable_baselines3/common/envs/identity_env.py @@ -4,7 +4,7 @@ from gymnasium import Env, Space from gym.spaces import Box, Discrete, MultiBinary, MultiDiscrete -from stable_baselines3.common.type_aliases import gymnasium as gym26ResetReturn, Gym26StepReturn +from stable_baselines3.common.type_aliases import Gym26ResetReturn, Gym26StepReturn class IdentityEnv(Env): diff --git a/stable_baselines3/common/envs/multi_input_envs.py b/stable_baselines3/common/envs/multi_input_envs.py index 8aa8612f1..f4a275f06 100644 --- a/stable_baselines3/common/envs/multi_input_envs.py +++ b/stable_baselines3/common/envs/multi_input_envs.py @@ -3,7 +3,7 @@ import gymnasium as gym import numpy as np -from stable_baselines3.common.type_aliases import gymnasium as gym26StepReturn +from stable_baselines3.common.type_aliases import Gym26StepReturn class SimpleMultiObsEnv(gym.Env): diff --git a/stable_baselines3/common/monitor.py b/stable_baselines3/common/monitor.py index 7efb502da..83204f68d 100644 --- a/stable_baselines3/common/monitor.py +++ b/stable_baselines3/common/monitor.py @@ -11,7 +11,7 @@ import numpy as np import pandas -from stable_baselines3.common.type_aliases import gymnasium as gym26ResetReturn, Gym26StepReturn +from stable_baselines3.common.type_aliases import Gym26ResetReturn, Gym26StepReturn class Monitor(gym.Wrapper):