fixe tests of test_env_after_learn

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()