From 9d4fb76ef01142eefde82132fd8491fbe87537c6 Mon Sep 17 00:00:00 2001 From: Peter Zhokhov Date: Tue, 6 Nov 2018 09:58:43 -0800 Subject: [PATCH] making num_envs and video length smaller in test_video_recorder to prevent hanging on travis --- baselines/common/vec_env/test_video_recorder.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/baselines/common/vec_env/test_video_recorder.py b/baselines/common/vec_env/test_video_recorder.py index f78fb59..363404a 100644 --- a/baselines/common/vec_env/test_video_recorder.py +++ b/baselines/common/vec_env/test_video_recorder.py @@ -14,8 +14,8 @@ from .subproc_vec_env import SubprocVecEnv from .vec_video_recorder import VecVideoRecorder @pytest.mark.parametrize('klass', (DummyVecEnv, ShmemVecEnv, SubprocVecEnv)) -@pytest.mark.parametrize('num_envs', (1, 16)) -@pytest.mark.parametrize('video_length', (10, 500)) +@pytest.mark.parametrize('num_envs', (1, 4)) +@pytest.mark.parametrize('video_length', (10, 100)) @pytest.mark.parametrize('video_interval', (1, 50)) def test_video_recorder(klass, num_envs, video_length, video_interval): """