test

Cray-HPE · Sep 17, 2024 · d0ad40e · d0ad40e
1 parent eb209ce
commit d0ad40e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/cray/modules/bos/cli.py b/cray/modules/bos/cli.py
@@ -199,6 +199,7 @@ def setup_components_patch():
     option(
         '--clear-pending-state',
         is_flag=True,
+        callback=_opt_callback,
         help='Shortcut for --patch \'{ '
              '"staged_state": { '
                 '"session": "", "configuration": "", '
@@ -235,17 +236,18 @@ def _decorator(clear_pending_state, **kwargs):
 def modify_component_patch():
     """ Modifies update command for components """
     source_command = cli.commands['v2'].commands['components'].commands['update']
-    source_command.params.append(option(
+    option(
         '--clear-pending-state',
         is_flag=True,
+        callback=_opt_callback,
         help='Shortcut for --staged-state-session "" --staged-state-configuration "" '
              '--staged-state-boot-artifacts-initrd "" '
              '--staged-state-boot-artifacts-kernel-parameters "" '
              '--staged-state-boot-artifacts-kernel "" --desired-state-bss-token "" '
              '--desired-state-configuration "" --desired-state-boot-artifacts-initrd "" '
              '--desired-state-boot-artifacts-kernel-parameters "" '
              '--desired-state-boot-artifacts-kernel ""'
-    )(source_command))
+    )(source_command)
     source_command.callback = create_patch_compnent_shim(source_command.callback)