Update bittensor_cli/src/commands/stake/claim.py

Co-authored-by: Roman <[email protected]>

Author: ibraheem-abe

bittensor_cli/src/commands/stake/claim.py

@@ -82,9 +82,9 @@ async def set_claim_type(
     if claim_type is not None:
         new_type = claim_type
     else:
-        new_type = Prompt.ask(
-            "Select new root claim type", choices=["Swap", "Keep"], default=current_type
-        )
+new_type = claim_type if claim_type else Prompt.ask(
+    "Select new root claim type", choices=["Swap", "Keep"], default=current_type
+)
     if new_type == current_type:
         msg = f"Root claim type is already set to '{current_type}'. No change needed."
         console.print(f"[yellow]{msg}[/yellow]")