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.
This commit is contained in:
karajan1001
2022-08-09 17:50:48 +08:00
committed by Gao
parent 2ca1d6c7c1
commit d6f8e5a507
+1
View File
@@ -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())