Added block_hash specification

Author: thewhaleking

bittensor_cli/src/bittensor/subtensor_interface.py

@@ -1120,9 +1120,11 @@ async def get_delegates_by_netuid_light(
 
         return DelegateInfoLite.list_from_vec_u8(result)  # TODO this won't work yet
 
-    async def get_subnet_dynamic_info(self, netuid: int) -> Optional["DynamicInfo"]:
+    async def get_subnet_dynamic_info(
+        self, netuid: int, block_hash: Optional[str] = None
+    ) -> Optional["DynamicInfo"]:
         json = await self.substrate.rpc_request(
-            method="subnetInfo_getDynamicInfo", params=[netuid, None]
+            method="subnetInfo_getDynamicInfo", params=[netuid, block_hash]
         )
         subnets = DynamicInfo.from_vec_u8(json["result"])
         return subnets
@@ -1188,7 +1190,9 @@ async def multi_get_stake_for_coldkey_and_hotkey_on_netuid(
             for netuid in netuids
         ]
         batch_call = await self.substrate.query_multi(calls, block_hash=block_hash)
-        results: dict[str, dict[int, "Balance"]] = {hk_ss58: {} for hk_ss58 in hotkey_ss58s}
+        results: dict[str, dict[int, "Balance"]] = {
+            hk_ss58: {} for hk_ss58 in hotkey_ss58s
+        }
         for idx, item in enumerate(batch_call):
             hotkey_idx = idx // len(netuids)
             netuid_idx = idx % len(netuids)

bittensor_cli/src/commands/stake/stake.py

@@ -1255,12 +1255,19 @@ async def stake_add(
         hotkey_ss58_or_name = wallet.hotkey.ss58_address
         hotkeys_to_stake_to = [(None, hotkey_ss58_or_name)]
 
+    starting_chain_head = await subtensor.substrate.get_chain_head()
     all_dynamic_info, initial_stake_balances = await asyncio.gather(
-        asyncio.gather(*[subtensor.get_subnet_dynamic_info(x) for x in netuids]),
+        asyncio.gather(
+            *[
+                subtensor.get_subnet_dynamic_info(x, starting_chain_head)
+                for x in netuids
+            ]
+        ),
         subtensor.multi_get_stake_for_coldkey_and_hotkey_on_netuid(
             hotkey_ss58s=[x[1] for x in hotkeys_to_stake_to],
             coldkey_ss58=wallet.coldkeypub.ss58_address,
             netuids=netuids,
+            block_hash=starting_chain_head,
         ),
     )
     for hk_name, hk_ss58 in hotkeys_to_stake_to: