11import asyncio
22import copy
3- from datetime import datetime , timezone
43import ssl
4+ from datetime import datetime , timezone
55from functools import partial
66from typing import Optional , Any , Union , Iterable , TYPE_CHECKING
77
88import asyncstdlib as a
99import numpy as np
1010import scalecodec
1111from async_substrate_interface import AsyncSubstrateInterface
12+ from bittensor_commit_reveal import get_encrypted_commitment
1213from bittensor_wallet .utils import SS58_FORMAT
1314from numpy .typing import NDArray
1415from scalecodec import GenericCall
2728 decode_account_id ,
2829 DynamicInfo ,
2930)
30- from bittensor_commit_reveal import get_encrypted_commitment
3131from bittensor .core .chain_data .chain_identity import ChainIdentity
3232from bittensor .core .chain_data .delegate_info import DelegatedInfo
3333from bittensor .core .chain_data .utils import (
3838from bittensor .core .config import Config
3939from bittensor .core .errors import ChainError , SubstrateRequestException
4040from bittensor .core .extrinsics .asyncex .commit_reveal import commit_reveal_v3_extrinsic
41+ from bittensor .core .extrinsics .asyncex .move_stake import (
42+ transfer_stake_extrinsic ,
43+ swap_stake_extrinsic ,
44+ move_stake_extrinsic ,
45+ )
4146from bittensor .core .extrinsics .asyncex .registration import (
4247 burned_register_extrinsic ,
4348 register_extrinsic ,
4449 register_subnet_extrinsic ,
4550 set_subnet_identity_extrinsic ,
4651)
47- from bittensor .core .extrinsics .asyncex .move_stake import (
48- transfer_stake_extrinsic ,
49- swap_stake_extrinsic ,
50- move_stake_extrinsic ,
51- )
5252from bittensor .core .extrinsics .asyncex .root import (
5353 set_root_weights_extrinsic ,
5454 root_register_extrinsic ,
8484 decode_hex_identity_dict ,
8585 float_to_u64 ,
8686 format_error_message ,
87+ is_valid_ss58_address ,
8788 torch ,
8889 u16_normalized_float ,
8990 u64_normalized_float ,
@@ -1064,14 +1065,14 @@ async def get_all_commitments(
10641065 result [decode_account_id (id_ [0 ])] = decode_metadata (value )
10651066 return result
10661067
1067- async def get_revealed_commitment (
1068+ async def get_revealed_commitment_by_hotkey (
10681069 self ,
10691070 netuid : int ,
10701071 hotkey_ss58_address : Optional [str ] = None ,
10711072 block : Optional [int ] = None ,
10721073 block_hash : Optional [str ] = None ,
10731074 reuse_block : bool = False ,
1074- ) -> Optional [tuple [int , str ]]:
1075+ ) -> Optional [tuple [tuple [ int , str ], ... ]]:
10751076 """Returns hotkey related revealed commitment for a given netuid.
10761077
10771078 Arguments:
@@ -1084,6 +1085,9 @@ async def get_revealed_commitment(
10841085 Returns:
10851086 result (tuple[int, str): A tuple of reveal block and commitment message.
10861087 """
1088+ if not is_valid_ss58_address (address = hotkey_ss58_address ):
1089+ raise ValueError (f"Invalid ss58 address { hotkey_ss58_address } )
1090+
10871091 query = await self .query_module (
10881092 module = "Commitments" ,
10891093 name = "RevealedCommitments" ,
@@ -1092,15 +1096,50 @@ async def get_revealed_commitment(
10921096 block_hash = block_hash ,
10931097 reuse_block = reuse_block ,
10941098 )
1095- return decode_revealed_commitment (query )
1099+ if query is None :
1100+ return None
1101+ return tuple (decode_revealed_commitment (pair ) for pair in query )
1102+
1103+ async def get_revealed_commitment (
1104+ self ,
1105+ netuid : int ,
1106+ uid : int ,
1107+ block : Optional [int ] = None ,
1108+ ) -> Optional [tuple [tuple [int , str ], ...]]:
1109+ """Returns uid related revealed commitment for a given netuid.
1110+
1111+ Arguments:
1112+ netuid (int): The unique identifier of the subnetwork.
1113+ uid (int): The neuron uid to retrieve the commitment from.
1114+ block (Optional[int]): The block number to retrieve the commitment from. Default is ``None``.
1115+
1116+ Returns:
1117+ result (Optional[tuple[int, str]]: A tuple of reveal block and commitment message.
1118+
1119+ Example of result:
1120+ ( (12, "Alice message 1"), (152, "Alice message 2") )
1121+ ( (12, "Bob message 1"), (147, "Bob message 2") )
1122+ """
1123+ try :
1124+ meta_info = await self .get_metagraph_info (netuid , block = block )
1125+ if meta_info :
1126+ hotkey_ss58_address = meta_info .hotkeys [uid ]
1127+ else :
1128+ raise ValueError (f"Subnet with netuid { netuid } )
1129+ except IndexError :
1130+ raise ValueError (f"Subnet { netuid } { uid } )
1131+
1132+ return await self .get_revealed_commitment_by_hotkey (
1133+ netuid = netuid , hotkey_ss58_address = hotkey_ss58_address , block = block
1134+ )
10961135
10971136 async def get_all_revealed_commitments (
10981137 self ,
10991138 netuid : int ,
11001139 block : Optional [int ] = None ,
11011140 block_hash : Optional [str ] = None ,
11021141 reuse_block : bool = False ,
1103- ) -> dict [str , tuple [int , str ]]:
1142+ ) -> dict [str , tuple [tuple [ int , str ], ... ]]:
11041143 """Returns all revealed commitments for a given netuid.
11051144
11061145 Arguments:
@@ -1111,6 +1150,12 @@ async def get_all_revealed_commitments(
11111150
11121151 Returns:
11131152 result (dict): A dictionary of all revealed commitments in view {ss58_address: (reveal block, commitment message)}.
1153+
1154+ Example of result:
1155+ {
1156+ "5GrwvaEF5zXb26Fz9rcQpDWS57CtERHpNehXCPcNoHGKutQY": ( (12, "Alice message 1"), (152, "Alice message 2") ),
1157+ "5FHneW46xGXgs5mUiveU4sbTyGBzmstUspZC92UhjJM694ty": ( (12, "Bob message 1"), (147, "Bob message 2") ),
1158+ }
11141159 """
11151160 query = await self .query_map (
11161161 module = "Commitments" ,
@@ -1122,8 +1167,11 @@ async def get_all_revealed_commitments(
11221167 )
11231168
11241169 result = {}
1125- async for item in query :
1126- result .update (decode_revealed_commitment_with_hotkey (item ))
1170+ async for pair in query :
1171+ hotkey_ss58_address , commitment_message = (
1172+ decode_revealed_commitment_with_hotkey (pair )
1173+ )
1174+ result [hotkey_ss58_address ] = commitment_message
11271175 return result
11281176
11291177 async def get_current_weight_commit_info (
@@ -1639,6 +1687,38 @@ async def get_neuron_for_pubkey_and_subnet(
16391687 reuse_block = reuse_block ,
16401688 )
16411689
1690+ async def get_next_epoch_start_block (
1691+ self ,
1692+ netuid : int ,
1693+ block : Optional [int ] = None ,
1694+ block_hash : Optional [str ] = None ,
1695+ reuse_block : bool = False ,
1696+ ) -> Optional [int ]:
1697+ """
1698+ Calculates the first block number of the next epoch for the given subnet.
1699+
1700+ If `block` is not provided, the current chain block will be used. Epochs are
1701+ determined based on the subnet's tempo (i.e., blocks per epoch). The result
1702+ is the block number at which the next epoch will begin.
1703+
1704+ Args:
1705+ netuid (int): The unique identifier of the subnet.
1706+ block (Optional[int], optional): The reference block to calculate from.
1707+ If None, uses the current chain block height.
1708+ block_hash (Optional[int]): The blockchain block number at which to perform the query.
1709+ reuse_block (bool): Whether to reuse the last-used blockchain block hash.
1710+
1711+
1712+ Returns:
1713+ int: The block number at which the next epoch will start.
1714+ """
1715+ block_hash = await self .determine_block_hash (block , block_hash , reuse_block )
1716+ if not block_hash and reuse_block :
1717+ block_hash = self .substrate .last_block_hash
1718+ block = await self .substrate .get_block_number (block_hash = block_hash )
1719+ tempo = await self .tempo (netuid = netuid , block_hash = block_hash )
1720+ return (((block // tempo ) + 1 ) * tempo ) + 1 if tempo else None
1721+
16421722 async def get_owned_hotkeys (
16431723 self ,
16441724 coldkey_ss58 : str ,
@@ -3764,6 +3844,8 @@ async def set_weights(
37643844 wait_for_inclusion : bool = False ,
37653845 wait_for_finalization : bool = False ,
37663846 max_retries : int = 5 ,
3847+ block_time : float = 12.0 ,
3848+ period : int = 5 ,
37673849 ):
37683850 """
37693851 Sets the inter-neuronal weights for the specified neuron. This process involves specifying the influence or
@@ -3783,6 +3865,8 @@ async def set_weights(
37833865 wait_for_finalization (bool): Waits for the transaction to be finalized on the blockchain. Default is
37843866 ``False``.
37853867 max_retries (int): The number of maximum attempts to set weights. Default is ``5``.
3868+ block_time (float): The amount of seconds for block duration. Default is 12.0 seconds.
3869+ period (int, optional): The period in seconds to wait for extrinsic inclusion or finalization. Defaults to 5.
37863870
37873871 Returns:
37883872 tuple[bool, str]: ``True`` if the setting of weights is successful, False otherwise. And `msg`, a string
@@ -3831,6 +3915,7 @@ async def _blocks_weight_limit() -> bool:
38313915 version_key = version_key ,
38323916 wait_for_inclusion = wait_for_inclusion ,
38333917 wait_for_finalization = wait_for_finalization ,
3918+ block_time = block_time ,
38343919 )
38353920 retries += 1
38363921 return success , message
@@ -3856,6 +3941,7 @@ async def _blocks_weight_limit() -> bool:
38563941 version_key = version_key ,
38573942 wait_for_inclusion = wait_for_inclusion ,
38583943 wait_for_finalization = wait_for_finalization ,
3944+ period = period ,
38593945 )
38603946 except Exception as e :
38613947 logging .error (f"Error setting weights: { e } )
0 commit comments