add callZooFunc and change all callBigDlFunc to callZooFunc (intel-an…

…alytics#1793)
Le-Zheng · Nov 26, 2019 · 566f754 · 566f754
1 parent 254efdb
commit 566f754
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/python/orca/src/bigdl/orca/ray/util/raycontext.py b/python/orca/src/bigdl/orca/ray/util/raycontext.py
@@ -38,11 +38,11 @@ class JVMGuard():
     def registerPids(pids):
         import traceback
         try:
-            from bigdl.util.common import callBigDlFunc
+            from zoo.common.utils import callZooFunc
             import zoo
-            callBigDlFunc("float",
-                          "jvmGuardRegisterPids",
-                          pids)
+            callZooFunc("float",
+                        "jvmGuardRegisterPids",
+                        pids)
         except Exception as err:
             print(traceback.format_exc())
             print("Cannot sucessfully register pid into JVMGuard")