fixes to catch changes in gym

This commit is contained in:
Peter Zhokhov
2019-02-27 12:49:40 -08:00
parent 5b41c926c7
commit adc4388f6b

View File

@@ -1,6 +1,5 @@
import tensorflow as tf
import numpy as np
from gym.spaces import np_random
from baselines.common.vec_env.dummy_vec_env import DummyVecEnv
N_TRIALS = 10000
@@ -8,8 +7,6 @@ N_EPISODES = 100
def simple_test(env_fn, learn_fn, min_reward_fraction, n_trials=N_TRIALS):
np.random.seed(0)
np_random.seed(0)
env = DummyVecEnv([env_fn])