From 566f754533ff531e7864ab4cf528f3ec751fb664 Mon Sep 17 00:00:00 2001 From: Xin Qiu Date: Tue, 26 Nov 2019 13:52:40 +0800 Subject: [PATCH] add callZooFunc and change all callBigDlFunc to callZooFunc (#1793) --- python/orca/src/bigdl/orca/ray/util/raycontext.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python/orca/src/bigdl/orca/ray/util/raycontext.py b/python/orca/src/bigdl/orca/ray/util/raycontext.py index ea9df58ad9f..559aac30552 100755 --- 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")