From d6f8e5a507cb0be9558787cdf72ab88aa2289a3e Mon Sep 17 00:00:00 2001 From: karajan1001 Date: Mon, 8 Aug 2022 20:53:49 +0800 Subject: [PATCH] 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. --- tests/func/experiments/test_queue.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/func/experiments/test_queue.py b/tests/func/experiments/test_queue.py index 92259c31b..3618ccfe3 100644 --- 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())