Skip to content

Commit 351640f

Browse files
committed
Merge branch 'staging' into feat/roman/get-hyperparameters-in-alphabetical-order
2 parents 1af1b90 + 5295a0a commit 351640f

File tree

2 files changed

+51
-18
lines changed

2 files changed

+51
-18
lines changed

bittensor_cli/cli.py

Lines changed: 48 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1070,7 +1070,8 @@ def initialize_chain(
10701070
not_selected_networks = [net for net in network if net != network_]
10711071
if not_selected_networks:
10721072
console.print(
1073-
f"Networks not selected: [dark_orange]{', '.join(not_selected_networks)}[/dark_orange]"
1073+
f"Networks not selected: "
1074+
f"[{COLORS.G.ARG}]{', '.join(not_selected_networks)}[/{COLORS.G.ARG}]"
10741075
)
10751076

10761077
self.subtensor = SubtensorInterface(network_)
@@ -1353,8 +1354,8 @@ def set_config(
13531354
if n := args.get("network"):
13541355
if n in Constants.networks:
13551356
if not Confirm.ask(
1356-
f"You provided a network [dark_orange]{n}[/dark_orange] which is mapped to "
1357-
f"[dark_orange]{Constants.network_map[n]}[/dark_orange]\n"
1357+
f"You provided a network [{COLORS.G.ARG}]{n}[/{COLORS.G.ARG}] which is mapped to "
1358+
f"[{COLORS.G.ARG}]{Constants.network_map[n]}[/{COLORS.G.ARG}]\n"
13581359
"Do you want to continue?"
13591360
):
13601361
typer.Exit()
@@ -1369,14 +1370,14 @@ def set_config(
13691370
)
13701371
args["network"] = known_network
13711372
if not Confirm.ask(
1372-
f"You provided an endpoint [dark_orange]{n}[/dark_orange] which is mapped to "
1373-
f"[dark_orange]{known_network}[/dark_orange]\n"
1373+
f"You provided an endpoint [{COLORS.G.ARG}]{n}[/{COLORS.G.ARG}] which is mapped to "
1374+
f"[{COLORS.G.ARG}]{known_network}[/{COLORS.G.ARG}]\n"
13741375
"Do you want to continue?"
13751376
):
1376-
typer.Exit()
1377+
raise typer.Exit()
13771378
else:
13781379
if not Confirm.ask(
1379-
f"You provided a chain endpoint URL [dark_orange]{n}[/dark_orange]\n"
1380+
f"You provided a chain endpoint URL [{COLORS.G.ARG}]{n}[/{COLORS.G.ARG}]\n"
13801381
"Do you want to continue?"
13811382
):
13821383
raise typer.Exit()
@@ -1451,15 +1452,15 @@ def del_config(
14511452
for arg in args.keys():
14521453
if self.config.get(arg) is not None:
14531454
if Confirm.ask(
1454-
f"Do you want to clear the [dark_orange]{arg}[/dark_orange] config?"
1455+
f"Do you want to clear the [{COLORS.G.ARG}]{arg}[/{COLORS.G.ARG}] config?"
14551456
):
14561457
self.config[arg] = None
14571458
console.print(
1458-
f"Cleared [dark_orange]{arg}[/dark_orange] config and set to 'None'."
1459+
f"Cleared [{COLORS.G.ARG}]{arg}[/{COLORS.G.ARG}] config and set to 'None'."
14591460
)
14601461
else:
14611462
console.print(
1462-
f"Skipped clearing [dark_orange]{arg}[/dark_orange] config."
1463+
f"Skipped clearing [{COLORS.G.ARG}]{arg}[/{COLORS.G.ARG}] config."
14631464
)
14641465

14651466
else:
@@ -1468,19 +1469,21 @@ def del_config(
14681469
if should_clear:
14691470
if self.config.get(arg) is not None:
14701471
if Confirm.ask(
1471-
f"Do you want to clear the [dark_orange]{arg}[/dark_orange] [bold cyan]({self.config.get(arg)})[/bold cyan] config?"
1472+
f"Do you want to clear the [{COLORS.G.ARG}]{arg}[/{COLORS.G.ARG}]"
1473+
f" [bold cyan]({self.config.get(arg)})[/bold cyan] config?"
14721474
):
14731475
self.config[arg] = None
14741476
console.print(
1475-
f"Cleared [dark_orange]{arg}[/dark_orange] config and set to 'None'."
1477+
f"Cleared [{COLORS.G.ARG}]{arg}[/{COLORS.G.ARG}] config and set to 'None'."
14761478
)
14771479
else:
14781480
console.print(
1479-
f"Skipped clearing [dark_orange]{arg}[/dark_orange] config."
1481+
f"Skipped clearing [{COLORS.G.ARG}]{arg}[/{COLORS.G.ARG}] config."
14801482
)
14811483
else:
14821484
console.print(
1483-
f"No config set for [dark_orange]{arg}[/dark_orange]. Use `btcli config set` to set it."
1485+
f"No config set for [{COLORS.G.ARG}]{arg}[/{COLORS.G.ARG}]."
1486+
f" Use [{COLORS.G.ARG}]`btcli config set`[/{COLORS.G.ARG}] to set it."
14841487
)
14851488
with open(self.config_path, "w") as f:
14861489
safe_dump(self.config, f)
@@ -1492,7 +1495,7 @@ def get_config(self):
14921495
deprecated_configs = ["chain"]
14931496

14941497
table = Table(
1495-
Column("[bold white]Name", style="dark_orange"),
1498+
Column("[bold white]Name", style=f"{COLORS.G.ARG}"),
14961499
Column("[bold white]Value", style="gold1"),
14971500
Column("", style="medium_purple"),
14981501
box=box.SIMPLE_HEAD,
@@ -4609,17 +4612,44 @@ def sudo_set(
46094612
if param_name in ["alpha_high", "alpha_low"]:
46104613
if not prompt:
46114614
err_console.print(
4612-
"`alpha_high` and `alpha_low` values cannot be set with `--no-prompt`"
4615+
f"[{COLORS.SU.HYPERPARAM}]alpha_high[/{COLORS.SU.HYPERPARAM}] and "
4616+
f"[{COLORS.SU.HYPERPARAM}]alpha_low[/{COLORS.SU.HYPERPARAM}] "
4617+
f"values cannot be set with `--no-prompt`"
46134618
)
46144619
return False
46154620
param_name = "alpha_values"
46164621
low_val = FloatPrompt.ask(
4617-
"Enter the new value for [dark_orange]alpha_low[/dark_orange]"
4622+
f"Enter the new value for [{COLORS.G.ARG}]alpha_low[/{COLORS.G.ARG}]"
46184623
)
46194624
high_val = FloatPrompt.ask(
4620-
"Enter the new value for [dark_orange]alpha_high[/dark_orange]"
4625+
f"Enter the new value for [{COLORS.G.ARG}]alpha_high[/{COLORS.G.ARG}]"
46214626
)
46224627
param_value = f"{low_val},{high_val}"
4628+
if param_name == "yuma_version":
4629+
if not prompt:
4630+
err_console.print(
4631+
f"[{COLORS.SU.HYPERPARAM}]yuma_version[/{COLORS.SU.HYPERPARAM}]"
4632+
f" is set using a different hyperparameter, and thus cannot be set with `--no-prompt`"
4633+
)
4634+
return False
4635+
if Confirm.ask(
4636+
f"[{COLORS.SU.HYPERPARAM}]yuma_version[/{COLORS.SU.HYPERPARAM}] can only be used to toggle Yuma 3. "
4637+
f"Would you like to toggle Yuma 3?"
4638+
):
4639+
param_name = "yuma3_enabled"
4640+
question = Prompt.ask(
4641+
"Would to like to enable or disable Yuma 3?",
4642+
choices=["enable", "disable"],
4643+
)
4644+
param_value = "true" if question == "enable" else "false"
4645+
else:
4646+
return False
4647+
if param_name == "subnet_is_active":
4648+
err_console.print(
4649+
f"[{COLORS.SU.HYPERPARAM}]subnet_is_active[/{COLORS.SU.HYPERPARAM}] "
4650+
f"is set by using [{COLORS.G.ARG}]`btcli subnets start`[/{COLORS.G.ARG}] command."
4651+
)
4652+
return False
46234653

46244654
if not param_value:
46254655
if not prompt:

bittensor_cli/src/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -661,6 +661,8 @@ class WalletValidationTypes(Enum):
661661
"yuma3_enabled": ("sudo_set_yuma3_enabled", False),
662662
"alpha_sigmoid_steepness": ("sudo_set_alpha_sigmoid_steepness", True),
663663
"user_liquidity_enabled": ("toggle_user_liquidity", False),
664+
"bonds_reset_enabled": ("sudo_set_bonds_reset_enabled", False),
665+
"transfers_enabled": ("sudo_set_toggle_transfer", False),
664666
}
665667

666668
HYPERPARAMS_MODULE = {
@@ -747,6 +749,7 @@ class General(Gettable):
747749
NETUID = "#CBA880" # Tan
748750
NETUID_EXTRA = "#DDD5A9" # Light Khaki
749751
TEMPO = "#67A3A5" # Grayish Teal
752+
ARG = "dark_orange"
750753
# aliases
751754
CK = COLDKEY
752755
HK = HOTKEY

0 commit comments

Comments
 (0)