FEAT-modin-project#1821: Some improvements in Cloudray factory and stuff

Signed-off-by: Vasilij Litvinov <vasilij.n.litvinov@intel.com>
vnlitvinov · Jul 28, 2020 · 7087cec · 7087cec
1 parent 864b267
commit 7087cec
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/modin/data_management/factories.py b/modin/data_management/factories.py
@@ -241,8 +241,7 @@ def __getattr__(self, name):
                     except KeyError:
 
                         def wrap(*a, _original=getattr(self.__io_cls, name), **kw):
-                            a = tuple(self.__conn.deliver(x) for x in a)
-                            kw = {k: self.__conn.deliver(v) for k, v in kw.items()}
+                            a, kw = self.__conn.deliver(a, kw)
                             return _original(*a, **kw)
 
                         self.__wrappers[name] = wrap

diff --git a/modin/experimental/cloud/rpyc_proxy.py b/modin/experimental/cloud/rpyc_proxy.py
@@ -26,7 +26,6 @@ def _batch_loads(items):
 class WrappingConnection(rpyc.Connection):
     def __init__(self, *a, **kw):
         super().__init__(*a, **kw)
-        self._remote_pickle_loads = None
         self._remote_batch_loads = None
         self._remote_cls_cache = {}
     def __wrap(self, local_obj):
@@ -115,7 +114,6 @@ def _box(self, obj):
         return super()._box(obj)
 
     def _init_deliver(self):
-        self._remote_pickle_loads = self.modules["rpyc.lib.compat"].pickle.loads
         self._remote_batch_loads = self.modules["modin.experimental.cloud.rpyc_proxy"]._batch_loads