Code improvements

Author: ibraheem-opentensor

.gitignore

@@ -107,14 +107,14 @@ ipython_config.py
 # pyenv
 #   For a library or package, you might want to ignore these files since the code is
 #   intended to run in multiple environments; otherwise, check them in:
-# .python-version
+.python-version
 
 # pipenv
 #   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
 #   However, in case of collaboration, if having platform-specific dependencies or dependencies
 #   having no cross-platform support, pipenv may install dependencies that don't work, or not
 #   install all needed dependencies.
-#Pipfile.lock
+Pipfile.lock
 
 # PEP 582; used by e.g. github.com/David-OConnor/pyflow
 __pypackages__/
@@ -164,8 +164,6 @@ cython_debug/
 # These are backup files generated by rustfmt
 **/*.rs.bk
 
-.DS_Store
-
 # The cache for docker container dependency
 .cargo
 

src/commands/wallets.py

@@ -1150,7 +1150,7 @@ async def _filter_stake_info(stake_info: StakeInfo) -> bool:
 async def transfer(
     wallet: Wallet, subtensor: SubtensorInterface, destination: str, amount: float
 ):
-    # To:do - work out prompts to be passed through the cli
+    # TODO: - work out prompts to be passed through the cli
     """Transfer token of amount to destination."""
     await transfer_extrinsic(
         subtensor, wallet, destination, Balance.from_tao(amount), prompt=False
@@ -1300,7 +1300,7 @@ async def faucet(
     log_verbose: bool,
     max_successes: int = 3,
 ):
-    # To:do - work out prompts to be passed through the cli 
+    # TODO: - work out prompts to be passed through the cli 
     success = await run_faucet_extrinsic(
         subtensor,
         wallet,

src/tests/e2e_tests/test_wallet_creations.py

@@ -143,16 +143,18 @@ def extract_mnemonics_from_commands(output: str) -> Dict[str, Optional[str]]:
     mnemonics: Dict[str, Optional[str]] = {"coldkey": None, "hotkey": None}
     lines = output.splitlines()
 
-    # Regex pattern to capture the mnemonic
-    pattern = re.compile(r"btcli w regen_(coldkey|hotkey) --mnemonic ([a-z ]+)")
+    key_types = ["coldkey", "hotkey"]
+    command_prefix = "btcli w regen_"
 
     for line in lines:
         line = line.strip().lower()
-        match = pattern.search(line)
-        if match:
-            key_type = match.group(1)  # 'coldkey' or 'hotkey'
-            mnemonic_phrase = match.group(2).strip()
-            mnemonics[key_type] = mnemonic_phrase
+
+        if line.startswith(command_prefix):
+            for key_type in key_types:
+                if line.startswith(f"{command_prefix}{key_type} --mnemonic "):
+                    mnemonic_phrase = line.split("--mnemonic ")[1].strip()
+                    mnemonics[key_type] = mnemonic_phrase
+                    break
 
     return mnemonics
 

src/tests/e2e_tests/utils.py

@@ -155,7 +155,7 @@ def validate_wallet_inspect(
     delegates: List[Tuple[str, float, bool]],
     hotkeys_netuid: List[Tuple[str, str, float, bool]],
 ):
-    # To:do - Handle stake in Balance format as well
+    # TODO: Handle stake in Balance format as well
     """
     Validates the presence of specific coldkey, balance, delegates, and hotkeys/netuid in the wallet information.