mirror of
https://github.com/wassname/ray.git
synced 2026-07-10 01:55:50 +08:00
[ray_client]: Insert decorators into the real ray module to allow for client mode (#13031)
This commit is contained in:
@@ -16,18 +16,12 @@ from ray.test_utils import wait_for_condition
|
||||
from ray.test_utils import wait_for_pid_to_exit
|
||||
from ray.tests.client_test_utils import create_remote_signal_actor
|
||||
|
||||
if client_test_enabled():
|
||||
from ray.experimental.client import ray
|
||||
else:
|
||||
import ray
|
||||
import ray
|
||||
# NOTE: We have to import setproctitle after ray because we bundle setproctitle
|
||||
# with ray.
|
||||
import setproctitle # noqa
|
||||
|
||||
|
||||
@pytest.mark.skipif(
|
||||
client_test_enabled(),
|
||||
reason="defining early, no ray package injection yet")
|
||||
def test_caching_actors(shutdown_only):
|
||||
# Test defining actors before ray.init() has been called.
|
||||
|
||||
@@ -705,7 +699,6 @@ def test_options_num_returns(ray_start_regular_shared):
|
||||
assert ray.get([obj1, obj2]) == [1, 2]
|
||||
|
||||
|
||||
@pytest.mark.skipif(client_test_enabled(), reason="remote args")
|
||||
def test_options_name(ray_start_regular_shared):
|
||||
@ray.remote
|
||||
class Foo:
|
||||
|
||||
Reference in New Issue
Block a user