From 62bde9a970d0ef1213de6fb77cf3c0105acf88cf Mon Sep 17 00:00:00 2001 From: Antonin RAFFIN Date: Tue, 5 May 2020 16:30:26 +0200 Subject: [PATCH] Fix import --- stable_baselines3/common/env_checker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stable_baselines3/common/env_checker.py b/stable_baselines3/common/env_checker.py index 751b204..6a7f781 100644 --- a/stable_baselines3/common/env_checker.py +++ b/stable_baselines3/common/env_checker.py @@ -5,7 +5,7 @@ import gym from gym import spaces import numpy as np -from stable_baselines.common.vec_env import DummyVecEnv, VecCheckNan +from stable_baselines3.common.vec_env import DummyVecEnv, VecCheckNan def _enforce_array_obs(observation_space: spaces.Space) -> bool: