13 lines
191 B
Python
13 lines
191 B
Python
import pytest
|
|
|
|
import ray
|
|
from ray.cluster_utils import Cluster
|
|
|
|
|
|
@pytest.fixture
|
|
def ray_start_cluster():
|
|
cluster = Cluster()
|
|
yield cluster
|
|
ray.shutdown()
|
|
cluster.shutdown()
|