Clean up a bit.

bittensor/commands/stake/revoke_children.py

@@ -18,13 +18,12 @@
 
 import argparse
 import re
-from typing import Union, Tuple
-import numpy as np
-from numpy.typing import NDArray
+from typing import Tuple, List
 from rich.prompt import Confirm, Prompt
 
 import bittensor
 from .. import defaults, GetChildrenCommand
+from ...utils import is_valid_ss58_address
 
 console = bittensor.__console__
 
@@ -84,9 +83,13 @@ def _run(cli: "bittensor.cli", subtensor: "bittensor.subtensor"):
         # Parse from strings
         netuid = cli.config.netuid
 
-        children = np.array(
-            [str(x) for x in re.split(r"[ ,]+", cli.config.children)], dtype=str
-        )
+        children = re.split(r"[ ,]+", cli.config.children.strip())
+
+        # Validate children SS58 addresses
+        for child in children:
+            if not is_valid_ss58_address(child):
+                console.print(f":cross_mark:[red] Invalid SS58 address: {child}[/red]")
+                return
 
         success, message = RevokeChildrenCommand.do_revoke_children_multiple(
             subtensor=subtensor,
@@ -152,7 +155,7 @@ def do_revoke_children_multiple(
         subtensor: "bittensor.subtensor",
         wallet: "bittensor.wallet",
         hotkey: str,
-        children: Union[NDArray[str], list],
+        children: List[str],
         netuid: int,
         wait_for_inclusion: bool = True,
         wait_for_finalization: bool = False,
@@ -168,7 +171,7 @@ def do_revoke_children_multiple(
                 Bittensor wallet object.
             hotkey (str):
                 Parent hotkey.
-            children (np.ndarray):
+            children (List[str]):
                 Children hotkeys.
             netuid (int):
                 Unique identifier of for the subnet.

bittensor/commands/stake/set_child.py

@@ -23,7 +23,7 @@
 
 import bittensor
 from .. import defaults, GetChildrenCommand  # type: ignore
-from ...utils.formatting import float_to_u64
+from ...utils.formatting import float_to_u64, normalize_u64_values
 
 console = bittensor.__console__
 
@@ -221,6 +221,7 @@ def do_set_child_singular(
             try:
                 # prepare values for emmit
                 proportion = float_to_u64(proportion)
+                proportion = normalize_u64_values([proportion])[0]
 
                 call_module = "SubtensorModule"
                 call_function = "set_child_singular"

bittensor/commands/stake/set_children.py

@@ -17,15 +17,19 @@
 
 import argparse
 import re
-from typing import Union
+from typing import Union, List
 from rich.prompt import Confirm
 from numpy.typing import NDArray
 import numpy as np
 from rich.prompt import Prompt
 from typing import Tuple
 import bittensor
 from .. import defaults, GetChildrenCommand  # type: ignore
-from ...utils.formatting import float_to_u64, normalize_u64_values
+from ...utils.formatting import (
+    float_to_u64,
+    normalize_u64_values,
+    is_valid_ss58_address,
+)
 
 console = bittensor.__console__
 
@@ -93,15 +97,16 @@ def _run(cli: "bittensor.cli", subtensor: "bittensor.subtensor"):
         # Parse from strings
         netuid = cli.config.netuid
 
-        proportions = np.array(
-            [float(x) for x in re.split(r"[ ,]+", cli.config.proportions)],
-            dtype=np.float32,
-        )
-        children = np.array(
-            [str(x) for x in re.split(r"[ ,]+", cli.config.children)], dtype=str
-        )
+        proportions = [float(x) for x in re.split(r"[ ,]+", cli.config.proportions)]
+        children = [str(x) for x in re.split(r"[ ,]+", cli.config.children)]
+
+        # Validate children SS58 addresses
+        for child in children:
+            if not is_valid_ss58_address(child):
+                console.print(f":cross_mark:[red] Invalid SS58 address: {child}[/red]")
+                return
 
-        total_proposed = np.sum(proportions) + current_proportions
+        total_proposed = sum(proportions) + current_proportions
         if total_proposed > 1:
             raise ValueError(
                 f":cross_mark:[red] The sum of all proportions cannot be greater than 1. Proposed sum of proportions is {total_proposed}[/red]"
@@ -181,7 +186,7 @@ def do_set_children_multiple(
         subtensor: "bittensor.subtensor",
         wallet: "bittensor.wallet",
         hotkey: str,
-        children: Union[NDArray[str], list],
+        children: List[str],
         netuid: int,
         proportions: Union[NDArray[np.float32], list],
         wait_for_inclusion: bool = True,
@@ -198,7 +203,7 @@ def do_set_children_multiple(
                 Bittensor wallet object.
             hotkey (str):
                 Parent hotkey.
-            children (np.ndarray):
+            children (List[str]):
                 Children hotkeys.
             netuid (int):
                 Unique identifier of for the subnet.
@@ -241,34 +246,14 @@ def do_set_children_multiple(
                     else proportions
                 )
 
-                # Log initial proportions
-                bittensor.logging.info(f"Initial proportions: {proportions_val}")
-                bittensor.logging.info(
-                    f"Sum of initial proportions: {sum(proportions_val)}"
-                )
-
                 # Convert each proportion value to u64
                 proportions_val = [
                     float_to_u64(proportion) for proportion in proportions_val
                 ]
 
-                # Log after float_to_u64 conversion
-                bittensor.logging.info(
-                    f"Proportions after float_to_u64: {proportions_val}"
-                )
-                bittensor.logging.info(
-                    f"Sum after float_to_u64: {sum(proportions_val)}"
-                )
-
                 # Normalize the u64 values to ensure their sum equals u64::MAX
                 proportions_val = normalize_u64_values(proportions_val)
 
-                # Log after normalization
-                bittensor.logging.info(f"Normalized proportions: {proportions_val}")
-                bittensor.logging.info(
-                    f"Sum of normalized proportions: {sum(proportions_val)}"
-                )
-
                 children_with_proportions = list(zip(children, proportions_val))
 
                 call_module = "SubtensorModule"

bittensor/utils/formatting.py

@@ -1,6 +1,8 @@
 import math
 from typing import List
 
+import bittensor
+
 
 def get_human_readable(num, suffix="H"):
     for unit in ["", "K", "M", "G", "T", "P", "E", "Z"]:
@@ -101,3 +103,13 @@ def normalize_u64_values(values: List[int]) -> List[int]:
     ), f"Sum of normalized values ({sum(normalized)}) is not equal to u64::MAX ({u64_max})"
 
     return normalized
+
+
+def is_valid_ss58_address(address: str) -> bool:
+    """
+    Validate that the hotkey address input str is a valid ss58 address.
+    """
+    try:
+        return bittensor.utils.ss58.is_valid_ss58_address(address)
+    except:
+        return False

tests/e2e_tests/subcommands/stake/test_set_revoke_child_hotkeys.py

@@ -96,7 +96,9 @@ def test_set_revoke_child(local_chain, capsys):
     )
 
     subtensor = bittensor.subtensor(network="ws://localhost:9945")
-    assert len(subtensor.get_children_info(netuid=1)) == 1, "failed to set child hotkey"
+    assert (
+        len(subtensor.get_children_info(netuid=1)[alice_keypair.ss58_address]) == 1
+    ), "failed to set child hotkey"
 
     output = capsys.readouterr().out
     assert "✅ Finalized" in output
@@ -222,13 +224,10 @@ def test_set_revoke_children(local_chain, capsys):
     )
 
     subtensor = bittensor.subtensor(network="ws://localhost:9945")
-    children_info = subtensor.get_children_info(netuid=1)
-    print("subtensor.get_children_info(netuid=1)", children_info)
-
-    # Get the first (and only) key in the dictionary
-    parent_key = next(iter(children_info))
+    assert (
+        len(subtensor.get_children_info(netuid=1)[alice_keypair.ss58_address]) == 2
+    ), "failed to set children hotkeys"
 
-    assert len(children_info[parent_key]) == 2, "failed to set children hotkeys"
     output = capsys.readouterr().out
     assert "✅ Finalized" in output