mirror of
https://github.com/saymrwulf/stable-baselines3.git
synced 2026-05-16 21:10:08 +00:00
Fix import in tests
This commit is contained in:
parent
62bde9a970
commit
04d85ac2e2
1 changed files with 1 additions and 1 deletions
|
|
@ -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):
|
||||
|
|
|
|||
Loading…
Reference in a new issue