[CI] Use rerunfailures instead of flaky (#6061)

* Use rerunfailures instead of flaky

* Lint
This commit is contained in:
Simon Mo
2019-11-01 13:59:03 -07:00
committed by GitHub
parent 6166ef3e09
commit c8d7065bf3
2 changed files with 9 additions and 10 deletions
+1 -2
View File
@@ -2,7 +2,6 @@ from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from flaky import flaky
import logging
import pytest
import time
@@ -155,7 +154,7 @@ def test_heartbeats_single(ray_start_cluster_head):
ray.get(work_handle)
@flaky(max_runs=4)
@pytest.mark.flaky(reruns=4)
def test_heartbeats_cluster(ray_start_cluster_head):
"""Unit test for `Cluster.wait_for_nodes`.