From 555c23afa383bf5afe30d9b6a3e64890993e13a3 Mon Sep 17 00:00:00 2001 From: Benjamin Himes Date: Thu, 7 Nov 2024 16:10:26 +0200 Subject: [PATCH] Better handle incorrect file path for wallets. --- bittensor_cli/src/bittensor/utils.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/bittensor_cli/src/bittensor/utils.py b/bittensor_cli/src/bittensor/utils.py index b6b05baae..2a7f2b89e 100644 --- a/bittensor_cli/src/bittensor/utils.py +++ b/bittensor_cli/src/bittensor/utils.py @@ -238,11 +238,14 @@ def get_hotkey_wallets_for_wallet( def get_coldkey_wallets_for_path(path: str) -> list[Wallet]: """Gets all wallets with coldkeys from a given path""" wallet_path = Path(path).expanduser() - wallets = [ - Wallet(name=directory.name, path=path) - for directory in wallet_path.iterdir() - if directory.is_dir() - ] + try: + wallets = [ + Wallet(name=directory.name, path=path) + for directory in wallet_path.iterdir() + if directory.is_dir() + ] + except FileNotFoundError: + wallets = [] return wallets