bug: flux module load/unload does not work in python

Signed-off-by: vsoch <[email protected]>

Author: vsoch

ensemble/members/flux/minicluster.py

@@ -68,7 +68,7 @@ def grow(self, rule, record=None):
         # For now use the ensemble type as the name
         # TODO this needs to be caught and decided upon - retry?
         response = self.client.action_request(
-            name=name, namespace=namespace, action="grow", payload={}
+            member=name, namespace=namespace, action="grow", payload={}
         )
         print(response)
 
@@ -81,7 +81,7 @@ def shrink(self, rule, record=None):
         namespace = self.options["namespace"]
 
         response = self.client.action_request(
-            name=name, namespace=namespace, action="shrink", payload={}
+            member=name, namespace=namespace, action="shrink", payload={}
         )
         print(response)
 

ensemble/members/flux/queue.py

@@ -1,3 +1,4 @@
+import os
 import shlex
 import sys
 import time
@@ -308,10 +309,9 @@ def setup_flux_heartbeat(self):
             return
 
         # Customize the heartbeat duration by reloading the module
-        self.handle.rpc("module.remove", {"name": "heartbeat"}).get()
-        self.handle.rpc(
-            "module.load", {"path": "heartbeat", "args": [f"period={self.cfg.heartbeat}s"]}
-        ).get()
+        # We call to system flux since python can error
+        assert os.system("flux module remove heartbeat") == 0
+        assert os.system(f"flux module load heartbeat period={self.cfg.heartbeat}s") == 0
 
         def heartbeat_callback(handle, msg_handler, msg, arg):
             print("💗 HEARTBEAT")