Skip to content

Commit 498a1f7

Browse files
authored
Merge pull request #582 from opentensor/fix/thewhaleking/handle-int-valueerror-better
2 parents a533d1d + 1ce3344 commit 498a1f7

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

bittensor_cli/cli.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -410,10 +410,15 @@ def get_optional_netuid(netuid: Optional[int], all_netuids: bool) -> Optional[in
410410
)
411411
if answer is None:
412412
return None
413+
answer = answer.strip()
413414
if answer.lower() == "all":
414415
return None
415416
else:
416-
return int(answer)
417+
try:
418+
return int(answer)
419+
except ValueError:
420+
err_console.print(f"Invalid netuid: {answer}")
421+
return get_optional_netuid(None, False)
417422
else:
418423
return netuid
419424

0 commit comments

Comments
 (0)