Fix no cpus test (#5009)

This commit is contained in:
Joey Jiang
2019-06-21 17:08:25 +08:00
committed by Hao Chen
parent 3b23d94cb8
commit a7f84b536f
2 changed files with 9 additions and 1 deletions
+8
View File
@@ -56,6 +56,14 @@ def _ray_start(**kwargs):
ray.shutdown()
# The following fixture will start ray with 0 cpu.
@pytest.fixture
def ray_start_no_cpu(request):
param = getattr(request, "param", {})
with _ray_start(num_cpus=0, **param) as res:
yield res
# The following fixture will start ray with 1 cpu.
@pytest.fixture
def ray_start_regular(request):