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

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

Author: vsoch

ensemble/members/flux/minicluster.py

@@ -61,28 +61,16 @@ def grow(self, rule, record=None):
         """
         Request to the API to grow the MiniCluster
         """
-        # We need to send over the name and namespace
         name = self.options["name"]
-        namespace = self.options["namespace"]
-
-        # 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={}
-        )
+        response = self.client.action_request(member=name, action="grow", payload={})
         print(response)
 
     def shrink(self, rule, record=None):
         """
         Request to the API to shrink the MiniCluster
         """
-        # We need to send over the name and namespace
         name = self.options["name"]
-        namespace = self.options["namespace"]
-
-        response = self.client.action_request(
-            name=name, namespace=namespace, action="shrink", payload={}
-        )
+        response = self.client.action_request(member=name, action="shrink", payload={})
         print(response)
 
     @property

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")