fix callBigDLFunc return a Int while the true return value from java …

…is a byte array. (intel-analytics#3111)
Le-Zheng · Mar 23, 2021 · 1ccef7f · 1ccef7f
1 parent e273ae0
commit 1ccef7f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/dllib/src/bigdl/dllib/utils/common.py b/python/dllib/src/bigdl/dllib/utils/common.py
@@ -634,8 +634,8 @@ def _java2py(gateway, r, encoding="bytes"):
             except Py4JJavaError:
                 pass  # not pickable
 
-    if isinstance(r, (bytearray, bytes)):
-        r = PickleSerializer().loads(bytes(r), encoding=encoding)
+        if isinstance(r, (bytearray, bytes)):
+            r = PickleSerializer().loads(bytes(r), encoding=encoding)
     return r