FEAT-modin-project#1821: Try making a faster connection

Signed-off-by: Vasilij Litvinov <vasilij.n.litvinov@intel.com>
vnlitvinov · Jul 28, 2020 · 3008671 · 3008671
1 parent 35f06f7
commit 3008671
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/modin/experimental/cloud/connection.py b/modin/experimental/cloud/connection.py
@@ -106,11 +106,12 @@ def __try_connect(self):
         from .rpyc_proxy import WrappingService
 
         try:
-            self.__connection = rpyc.connect(
-                "127.0.0.1",
-                self.rpyc_port,
+            stream = rpyc.SocketStream.connect(
+                host="127.0.0.1", port=self.rpyc_port, nodelay=True, keepalive=True
+            )
+            self.__connection = rpyc.connect_stream(
+                stream,
                 WrappingService,
-                keepalive=True,
                 config={"sync_request_timeout": RPYC_REQUEST_TIMEOUT},
             )
         except (ConnectionRefusedError, EOFError):