Fix import

This commit is contained in:
Antonin RAFFIN 2020-05-05 16:30:26 +02:00
parent 2c34a4d694
commit 62bde9a970

View file

@ -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: