mirror of
https://github.com/wassname/ray.git
synced 2026-07-18 12:40:56 +08:00
Enable even more new scheduler tests (#12096)
This commit is contained in:
@@ -10,9 +10,9 @@ import ray
|
||||
import ray.test_utils
|
||||
from ray.core.generated import common_pb2
|
||||
from ray.core.generated import node_manager_pb2, node_manager_pb2_grpc
|
||||
from ray.test_utils import (wait_for_condition, wait_for_pid_to_exit,
|
||||
run_string_as_driver,
|
||||
run_string_as_driver_nonblocking)
|
||||
from ray.test_utils import (
|
||||
wait_for_condition, wait_for_pid_to_exit, run_string_as_driver,
|
||||
run_string_as_driver_nonblocking, new_scheduler_enabled)
|
||||
|
||||
|
||||
def get_workers():
|
||||
@@ -207,6 +207,7 @@ def test_worker_capping_run_chained_tasks(shutdown_only):
|
||||
assert len(get_workers()) == 2
|
||||
|
||||
|
||||
@pytest.mark.skipif(new_scheduler_enabled(), reason="fails")
|
||||
def test_worker_capping_fifo(shutdown_only):
|
||||
# Start 2 initial workers by setting num_cpus to 2.
|
||||
info = ray.init(num_cpus=2)
|
||||
@@ -250,6 +251,7 @@ ray.shutdown()
|
||||
assert worker2.pid == get_workers()[0].pid
|
||||
|
||||
|
||||
@pytest.mark.skipif(new_scheduler_enabled(), reason="raylet hang 100% cpu")
|
||||
def test_worker_registration_failure_after_driver_exit(shutdown_only):
|
||||
info = ray.init(num_cpus=1)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user