Update tests/func/experiments/test_queue.py

Author: skshetry

tests/func/experiments/test_queue.py

@@ -64,7 +64,7 @@ def test_celery_logs(
     assert "failed to reproduce 'failed-copy-file'" in captured.out
 
 
-@pytest.mark.flaky(rerun=3)
+@pytest.mark.flaky(max_runs=3, min_passes=1)
 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())