From d0ad40e80acd16650b9e9e680741fe746e5f2dcc Mon Sep 17 00:00:00 2001 From: "Mitch Harding (the weird one)" Date: Tue, 17 Sep 2024 15:55:19 -0400 Subject: [PATCH] test --- cray/modules/bos/cli.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/cray/modules/bos/cli.py b/cray/modules/bos/cli.py index ce6fd32..76eaf97 100644 --- 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,9 +236,10 @@ 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 "" ' @@ -245,7 +247,7 @@ def modify_component_patch(): '--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)