Add flaky to a queue related test.

This is a flaky, the queue tasks status might remain "Running" even if it had already been completed, beside sometimes I also noticed it raise a pytest fixture error.
giannipablo · Aug 9, 2022 · d6f8e5a · d6f8e5a
1 parent 2ca1d6c
commit d6f8e5a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/func/experiments/test_queue.py b/tests/func/experiments/test_queue.py
@@ -64,6 +64,7 @@ def test_celery_logs(
     assert "failed to reproduce 'failed-copy-file'" in captured.out
 
 
+@pytest.mark.flaky(rerun=3)
 def test_queue_remove_done(dvc, failed_tasks, success_tasks):
     assert len(dvc.experiments.celery_queue.failed_stash) == 3
     status = to_dict(dvc.experiments.celery_queue.status())