tests: mark test_extend_branch as flaky (iterative#4718)

efiop · Oct 14, 2020 · 4aa8fd0 · 4aa8fd0
1 parent 3cebd98
commit 4aa8fd0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/func/experiments/test_experiments.py b/tests/func/experiments/test_experiments.py
@@ -1,6 +1,7 @@
 from textwrap import dedent
 
 import pytest
+from flaky.flaky_decorator import flaky
 from funcy import first
 
 from dvc.utils.serialize import PythonFileCorruptedError
@@ -205,6 +206,8 @@ def test_update_py_params(tmp_dir, scm, dvc):
         dvc.experiments.run(stage.addressing, params=["params.py:INT=2a"])
 
 
+# See: https://github.com/iterative/dvc/issues/4717
+@flaky(max_runs=3, min_passes=1)
 def test_extend_branch(tmp_dir, scm, dvc):
     tmp_dir.gen("copy.py", COPY_SCRIPT)
     tmp_dir.gen("params.yaml", "foo: 1")