- changed imports

RaikoPipe · Oct 31, 2022 · c4868b2 · c4868b2
1 parent 52cc1f1
commit c4868b2
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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):