Fixing test flakiness (#2242)

Co-authored-by: ziyiwu9494 <[email protected]>
rlworkgroup · Apr 13, 2021 · 90b6090 · 90b6090
1 parent 6700814
commit 90b6090
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/garage/sampler/test_multiprocessing_sampler.py b/tests/garage/sampler/test_multiprocessing_sampler.py
@@ -50,7 +50,7 @@ def test_obtain_samples():
     env.close()
 
 
-@pytest.mark.flaky
+@pytest.mark.flaky(max_runs=3)
 @pytest.mark.timeout(10)
 def test_update_envs_env_update():
     max_episode_length = 16