gaspersic/baselines
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixes to catch changes in gym

Browse Source
This commit is contained in:
Peter Zhokhov
2019-02-27 12:49:40 -08:00
parent 5b41c926c7
commit adc4388f6b
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
baselines/common/tests/util.py
View File

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