keep pilot RPCs local

radical-cybertools · Sep 22, 2023 · 48d2df4 · 48d2df4
1 parent d1ef2b1
commit 48d2df4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/radical/pilot/agent/agent_0.py b/src/radical/pilot/agent/agent_0.py
@@ -214,7 +214,8 @@ def initialize(self):
                                  'export PATH=%s'
                                  %  os.environ.get('PATH', '')]
                    }
-        self.rpc('prepare_env', env_name='rp', env_spec=env_spec)
+        self.rpc('prepare_env', env_name='rp', env_spec=env_spec,
+                                addr=self._pid)
 
         # start any services if they are requested
         self._start_services()

diff --git a/src/radical/pilot/utils/component.py b/src/radical/pilot/utils/component.py
@@ -467,7 +467,7 @@ def register_rpc_handler(self, cmd, handler, addr=None):
 
     # --------------------------------------------------------------------------
     #
-    def rpc(self, cmd, *args, **kwargs):
+    def rpc(self, cmd, addr=None, *args, **kwargs):
         '''Remote procedure call.
 
         Send am RPC command and arguments to the control pubsub and wait for the
@@ -479,7 +479,8 @@ def rpc(self, cmd, *args, **kwargs):
 
         rpc_id  = ru.generate_id('%s.rpc' % self._uid)
         rpc_req = RPCRequestMessage(uid=rpc_id, cmd=cmd,
-                                    args=args, kwargs=kwargs)
+                                    args=args, kwargs=kwargs,
+                                    addr=addr)
 
         self._rpc_reqs[rpc_id] = {
                 'req': rpc_req,