Cleanup

Author: ibraheem-abe

bittensor_cli/cli.py

@@ -3234,10 +3234,8 @@ def stake_remove(
                         "Enter the [blue]hotkey[/blue] name or [blue]ss58 address[/blue] to unstake all from [dim](or enter 'all' to unstake from all hotkeys)[/dim]",
                         default=self.config.get("wallet_hotkey") or defaults.wallet.hotkey,
                     )
-                elif hotkey_ss58_address:
-                    hotkey_or_ss58 = hotkey_ss58_address
                 else:
-                    hotkey_or_ss58 = wallet_hotkey
+                    hotkey_or_ss58 = hotkey_ss58_address or wallet_hotkey
 
                 if is_valid_ss58_address(hotkey_or_ss58):
                     hotkey_ss58_address = hotkey_or_ss58

tests/e2e_tests/test_unstaking.py

@@ -97,7 +97,7 @@ def test_unstaking(local_chain, wallet_setup):
     )
     assert "✅ Registered subnetwork with netuid: 3" in result.stdout
 
-    # Register Bob in both subnets
+    # Register Bob in one subnet
     register_result = exec_command_bob(
         command="subnets",
         sub_command="register",