mirror of
https://github.com/wassname/ray.git
synced 2026-06-29 13:49:45 +08:00
[CI] Use rerunfailures instead of flaky (#6061)
* Use rerunfailures instead of flaky * Lint
This commit is contained in:
@@ -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`.
|
||||
|
||||
|
||||
Reference in New Issue
Block a user