bumped supersuit to 3.0.1

Farama-Foundation · Jul 8, 2021 · d43fcb7 · d43fcb7
1 parent f554278
commit d43fcb7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/supersuit/__init__.py b/supersuit/__init__.py
@@ -13,7 +13,7 @@
 from .aec_vector import vectorize_aec_env_v0 # NOQA
 
 
-__version__ = "3.0.0"
+__version__ = "3.0.1"
 
 
 import warnings

diff --git a/supersuit/vector/vector_constructors.py b/supersuit/vector/vector_constructors.py
@@ -51,10 +51,10 @@ def concat_vec_envs_v0(vec_env, num_vec_envs, num_cpus=0, base_class='gym'):
     if base_class == "gym":
         return vec_env
     elif base_class == "stable_baselines":
-        from .vector.sb_vector_wrapper import SBVecEnvWrapper
+        from .sb_vector_wrapper import SBVecEnvWrapper
         return SBVecEnvWrapper(vec_env)
     elif base_class == "stable_baselines3":
-        from .vector.sb3_vector_wrapper import SB3VecEnvWrapper
+        from .sb3_vector_wrapper import SB3VecEnvWrapper
         return SB3VecEnvWrapper(vec_env)
     else:
         raise ValueError("supersuit_vec_env only supports 'gym', 'stable_baselines', and 'stable_baselines3' for its base_class")