From 59a7ffb84d9cfaa388ec9577b8cf089d66006402 Mon Sep 17 00:00:00 2001 From: Peter Zhokhov Date: Mon, 10 Sep 2018 10:32:42 -0700 Subject: [PATCH] fixe tests of test_env_after_learn --- baselines/common/tests/test_env_after_learn.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/baselines/common/tests/test_env_after_learn.py b/baselines/common/tests/test_env_after_learn.py index 6b0890a..79c2448 100644 --- a/baselines/common/tests/test_env_after_learn.py +++ b/baselines/common/tests/test_env_after_learn.py @@ -12,17 +12,17 @@ algos = ['a2c', 'acer', 'acktr', 'deepq', 'ppo2', 'trpo_mpi'] @pytest.mark.parametrize('algo', algos) def test_env_after_learn(algo): def make_env(): - env = gym.make('PongNoFrameskip-v4') + # acktr requires too much RAM, fails on travis + env = gym.make('CartPole-v1' if algo == 'acktr' else 'PongNoFrameskip-v4') return env make_session(make_default=True, graph=tf.Graph()) env = SubprocVecEnv([make_env]) learn = get_learn_function(algo) - network = cnn(one_dim_bias=True) # Commenting out the following line resolves the issue, though crash happens at env.reset(). - learn(network=network, env=env, total_timesteps=0, load_path=None, seed=None) + learn(network='mlp', env=env, total_timesteps=0, load_path=None, seed=None) env.reset() env.close()