Fix import in tests

This commit is contained in:
Antonin RAFFIN 2020-05-05 16:32:08 +02:00
parent 62bde9a970
commit 04d85ac2e2

View file

@ -2,7 +2,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
class NanAndInfEnv(gym.Env):