diff --git a/bittensor_cli/src/bittensor/utils.py b/bittensor_cli/src/bittensor/utils.py index 97dcf0473..8611db3f8 100644 --- a/bittensor_cli/src/bittensor/utils.py +++ b/bittensor_cli/src/bittensor/utils.py @@ -1443,5 +1443,5 @@ def get_hotkey_pub_ss58(wallet: Wallet) -> str: """ try: return wallet.hotkeypub.ss58_address - except KeyFileError: + except (KeyFileError, AttributeError): return wallet.hotkey.ss58_address diff --git a/bittensor_cli/src/commands/wallets.py b/bittensor_cli/src/commands/wallets.py index 00132d3d9..ff37a50e7 100644 --- a/bittensor_cli/src/commands/wallets.py +++ b/bittensor_cli/src/commands/wallets.py @@ -452,13 +452,15 @@ async def wallet_create( "error": "", "data": None, } + if uri: try: keypair = Keypair.create_from_uri(uri) - wallet.set_coldkey(keypair=keypair, encrypt=False, overwrite=False) - wallet.set_coldkeypub(keypair=keypair, encrypt=False, overwrite=False) - wallet.set_hotkey(keypair=keypair, encrypt=False, overwrite=False) - wallet.set_coldkeypub(keypair=keypair, encrypt=False, overwrite=False) + wallet.set_coldkey(keypair=keypair, encrypt=False, overwrite=overwrite) + wallet.set_coldkeypub(keypair=keypair, encrypt=False, overwrite=overwrite) + wallet.set_hotkey(keypair=keypair, encrypt=False, overwrite=overwrite) + wallet.set_hotkeypub(keypair=keypair, encrypt=False, overwrite=overwrite) + wallet.set_coldkeypub(keypair=keypair, encrypt=False, overwrite=overwrite) output_dict["success"] = True output_dict["data"] = { "name": wallet.name,