test

Cray-HPE · Sep 18, 2024 · fbb91ad · fbb91ad
1 parent d87e35c
commit fbb91ad
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/cray/modules/bos/cli.py b/cray/modules/bos/cli.py
@@ -243,13 +243,19 @@ def _decorator(clear_desired_state, clear_staged_state, clear_pending_state, **k
             clear_staged_state["value"] = True
             clear_desired_state["value"] = True
         if clear_staged_state["value"]:
-            kwargs.update({
-                arg.replace('-','_'): { "name": arg, "value": "" }
-                for arg in STAGED_STATE_ARGS})
+            for arg in kwargs.values():
+                if arg["name"].find('staged_state') == 0:
+                    arg["value"] = ""
+            #kwargs.update({
+                #arg.replace('-','_'): { "name": arg, "value": "" }
+                #for arg in STAGED_STATE_ARGS})
         if clear_desired_state["value"]:
-            kwargs.update({
-                arg.replace('-','_'): { "name": arg, "value": "" }
-                for arg in DESIRED_STATE_ARGS})
+            for arg in kwargs.values():
+                if arg["name"].find('desired_state') == 0:
+                    arg["value"] = ""
+            #kwargs.update({
+                #arg.replace('-','_'): { "name": arg, "value": "" }
+                #for arg in DESIRED_STATE_ARGS})
         return func(**kwargs)
 
     return _decorator