diff --git a/bittensor/keyfile.py b/bittensor/keyfile.py index b5157cea4a..f1b2ad622e 100644 --- a/bittensor/keyfile.py +++ b/bittensor/keyfile.py @@ -281,7 +281,7 @@ def get_coldkey_password_from_environment(coldkey_name: str) -> Optional[str]: for env_name, env_value in os.environ.items() if (normalized_env_name := env_name.upper()).startswith("BT_COLD_PW_") } - return envs.get(f"BT_COLD_PW_{coldkey_name.upper()}") + return envs.get(f"BT_COLD_PW_{coldkey_name.replace('-', '_').upper()}") def decrypt_keyfile_data( diff --git a/tests/unit_tests/test_keyfile.py b/tests/unit_tests/test_keyfile.py index d20af809f9..8db105c3bd 100644 --- a/tests/unit_tests/test_keyfile.py +++ b/tests/unit_tests/test_keyfile.py @@ -613,7 +613,8 @@ def test_deserialize_keypair_from_keyfile_data(keyfile_setup_teardown): def test_get_coldkey_password_from_environment(monkeypatch): password_by_wallet = { "WALLET": "password", - "my_wallet": "password", + "my_wallet": "password2", + "my-wallet": "password2", } monkeypatch.setenv("bt_cold_pw_wallet", password_by_wallet["WALLET"])