1 files changed, 5 insertions, 6 deletions
diff --git a/rpc/rpc.py b/rpc/rpc.py
index ef8339a..51f8feb 100644
--- a/rpc/rpc.py
+++ b/rpc/rpc.py
@@ -12,18 +12,17 @@ load_dotenv(find_dotenv())
 class CortexRPC(object):
   def __init__(self, get_fn, set_fn, ready_fn):
     super().__init__() 
-    # self._get_fn = get_fn
-    # self._set_fn = set_fn
-    # self._ready_fn = ready_fn
+    self._get_fn = get_fn
+    self._set_fn = set_fn
     self.working = False
-    self.connect()
+    self.connect(ready_fn)
 
-  def connect(self):
+  def connect(self, ready_fn):
     print('Connecting to RPC on port {}'.format(os.getenv('RPC_PORT')))
     self.rpc = zerorpc.Server(self)
     self.rpc.bind('tcp://127.0.0.1:' + os.getenv('RPC_PORT'))
     gevent.spawn(self.connected)
-    gevent.spawn(self._ready_fn)
+    gevent.spawn(ready_fn)
     self.rpc.run()
 
   def connected(self):