diff --git a/python/orca/src/test/bigdl/orca/ray/integration/ray_on_yarn.py b/python/orca/src/test/bigdl/orca/ray/integration/ray_on_yarn.py index fcbf1aa59de..85193bf70c1 100644 --- a/python/orca/src/test/bigdl/orca/ray/integration/ray_on_yarn.py +++ b/python/orca/src/test/bigdl/orca/ray/integration/ray_on_yarn.py @@ -69,8 +69,8 @@ def network(self): actors = [TestRay.remote() for i in range(0, slave_num)] -print([ray.get(actor.hostname.remote()) for actor in actors]) -print([ray.get(actor.ip.remote()) for actor in actors]) -# print([ray.get(actor.network.remote()) for actor in actors]) +print(ray.get([actor.hostname.remote() for actor in actors])) +print(ray.get([actor.ip.remote() for actor in actors])) +# print(ray.get([actor.network.remote() for actor in actors])) ray_ctx.stop() diff --git a/python/orca/src/test/bigdl/orca/ray/integration/test_yarn_reinit_raycontext.py b/python/orca/src/test/bigdl/orca/ray/integration/test_yarn_reinit_raycontext.py index 6c38bf8e52f..e66ca2c9d17 100644 --- a/python/orca/src/test/bigdl/orca/ray/integration/test_yarn_reinit_raycontext.py +++ b/python/orca/src/test/bigdl/orca/ray/integration/test_yarn_reinit_raycontext.py @@ -47,13 +47,13 @@ def hostname(self): ray_ctx = RayContext(sc=sc, object_store_memory="2g") ray_ctx.init() actors = [TestRay.remote() for i in range(0, node_num)] -print([ray.get(actor.hostname.remote()) for actor in actors]) +print(ray.get([actor.hostname.remote() for actor in actors])) ray_ctx.stop() # repeat ray_ctx = RayContext(sc=sc, object_store_memory="1g") ray_ctx.init() actors = [TestRay.remote() for i in range(0, node_num)] -print([ray.get(actor.hostname.remote()) for actor in actors]) +print(ray.get([actor.hostname.remote() for actor in actors])) ray_ctx.stop() sc.stop() diff --git a/python/orca/src/test/bigdl/orca/ray/test_ray_on_local.py b/python/orca/src/test/bigdl/orca/ray/test_ray_on_local.py index 461b1203637..76e7bc3a8c2 100644 --- a/python/orca/src/test/bigdl/orca/ray/test_ray_on_local.py +++ b/python/orca/src/test/bigdl/orca/ray/test_ray_on_local.py @@ -42,7 +42,7 @@ def test_local(self): ray_ctx = RayContext(sc=sc, object_store_memory="1g") ray_ctx.init() actors = [TestRay.remote() for i in range(0, node_num)] - print([ray.get(actor.hostname.remote()) for actor in actors]) + print(ray.get([actor.hostname.remote() for actor in actors])) ray_ctx.stop() sc.stop() time.sleep(1) diff --git a/python/orca/src/test/bigdl/orca/ray/test_reinit_raycontext.py b/python/orca/src/test/bigdl/orca/ray/test_reinit_raycontext.py index 514955cbf48..a7aafcfc81c 100644 --- a/python/orca/src/test/bigdl/orca/ray/test_reinit_raycontext.py +++ b/python/orca/src/test/bigdl/orca/ray/test_reinit_raycontext.py @@ -42,7 +42,7 @@ def test_local(self): ray_ctx = RayContext(sc=sc, object_store_memory="1g") ray_ctx.init() actors = [TestRay.remote() for i in range(0, node_num)] - print([ray.get(actor.hostname.remote()) for actor in actors]) + print(ray.get([actor.hostname.remote() for actor in actors])) ray_ctx.stop() time.sleep(3) # repeat @@ -50,7 +50,7 @@ def test_local(self): ray_ctx = RayContext(sc=sc, object_store_memory="1g") ray_ctx.init() actors = [TestRay.remote() for i in range(0, node_num)] - print([ray.get(actor.hostname.remote()) for actor in actors]) + print(ray.get([actor.hostname.remote() for actor in actors])) ray_ctx.stop() sc.stop() time.sleep(3)