Fix ray get in related unit tests (intel-analytics#1985)

Le-Zheng · Feb 14, 2020 · 7d3790e · 7d3790e
1 parent 35e6560
commit 7d3790e
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
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
@@ -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
@@ -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
@@ -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
@@ -42,15 +42,15 @@ 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
             print("-------------------first repeat begin!------------------")
             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)