Skip to content

add name and symbol fields to metagraph #2658

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 5 commits into from
Feb 11, 2025
Merged

add name and symbol fields to metagraph #2658

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
bc0a478
add name and symbol fields to metagraph
Feb 11, 2025
82ca010
Moves shared features to the Mixin, fixes AsyncSubtensor, changes `sy…
thewhaleking Feb 11, 2025
f58aad6
brings all fields from get_metagraph_info fields to the metagraph cla…
Feb 11, 2025
942f2b4
Add 0.018 as a constant
thewhaleking Feb 11, 2025
4f2b7f7
Merge pull request #2659 from opentensor/fix/thewhaleking/fix-async-t…
basfroman Feb 11, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
58 changes: 40 additions & 18 deletions bittensor/core/metagraph.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,7 @@


Tensor = Union["torch.nn.Parameter", NDArray]
ROOT_TAO_STAKES_WEIGHT = 0.018


METAGRAPH_STATE_DICT_NDARRAY_KEYS = [
Expand Down Expand Up @@ -258,6 +259,11 @@ class MetagraphMixin(ABC):
_dtype_registry = {"int64": np.int64, "float32": np.float32, "bool": bool}

# metagraph_info fields
name: str
symbol: str
network_registered_at: int
num_uids: int
max_uids: int
identities: list[Optional["ChainIdentity"]]
identity: Optional["SubnetIdentity"]
pruning_score: list[float]
Expand Down Expand Up @@ -537,6 +543,15 @@ def __init__(

metagraph = Metagraph(netuid=123, network="finney", lite=True, sync=True)
"""
self.lite = lite
self.subtensor = subtensor
self.should_sync = sync
self.netuid = netuid
self.network, self.chain_endpoint = determine_chain_endpoint_and_network(
network
)
self.neurons = []
self.axons: list[AxonInfo] = []

def __str__(self) -> str:
"""
Expand Down Expand Up @@ -937,6 +952,11 @@ def _apply_metagraph_info_mixin(self, metagraph_info: "MetagraphInfo"):
metagraph_info (MetagraphInfo): An instance of the MetagraphInfo class containing the data to be applied to
the current object.
"""
self.name = metagraph_info.name
self.symbol = metagraph_info.symbol
self.network_registered_at = metagraph_info.network_registered_at
self.num_uids = metagraph_info.num_uids
self.max_uids = metagraph_info.max_uids
self.identities = metagraph_info.identities
self.identity = metagraph_info.identity
self.pruning_score = metagraph_info.pruning_score
Expand Down Expand Up @@ -1043,10 +1063,6 @@ def __init__(
"""
BaseClass.__init__(self)
MetagraphMixin.__init__(self, netuid, network, lite, sync, subtensor)
self.netuid = netuid
self.network, self.chain_endpoint = determine_chain_endpoint_and_network(
network
)
self._dtype_registry = {
"int64": torch.int64,
"float32": torch.float32,
Expand Down Expand Up @@ -1107,10 +1123,6 @@ def __init__(
self.uids = torch.nn.Parameter(
torch.tensor([], dtype=torch.int64), requires_grad=False
)
self.axons: list[AxonInfo] = []
self.neurons = []
self.subtensor = subtensor
self.should_sync = sync
self.alpha_stake = torch.nn.Parameter(
torch.tensor([], dtype=torch.float32), requires_grad=False
)
Expand Down Expand Up @@ -1239,9 +1251,6 @@ def __init__(
self.tao_stake: Tensor = np.array([], dtype=np.int64)
self.stake: Tensor = np.array([], dtype=np.int64)
self.total_stake: Tensor = np.array([], dtype=np.int64)

self.axons: list[AxonInfo] = []
self.neurons = []
self.subtensor = subtensor
self.should_sync = sync

Expand Down Expand Up @@ -1343,7 +1352,7 @@ async def __aexit__(self, exc_type, exc_val, exc_tb):
async def sync(
self,
block: Optional[int] = None,
lite: bool = True,
lite: Optional[bool] = None,
subtensor: Optional["AsyncSubtensor"] = None,
):
"""
Expand All @@ -1354,8 +1363,9 @@ async def sync(
Args:
block (Optional[int]): A specific block number to synchronize with. If None, the metagraph syncs with the
latest block. This allows for historical analysis or specific state examination of the network.
lite (bool): If True, a lite version of the metagraph is used for quicker synchronization. This is
lite (Optional[bool]): If True, a lite version of the metagraph is used for quicker synchronization. This is
beneficial when full detail is not necessary, allowing for reduced computational and time overhead.
Defaults to `True`.
subtensor (Optional[bittensor.core.subtensor.Subtensor]): An instance of the subtensor class from Bittensor,
providing an interface to the underlying blockchain data. If provided, this instance is used for data
retrieval during synchronization.
Expand Down Expand Up @@ -1390,6 +1400,9 @@ async def sync(

metagraph.sync(block=history_block, lite=False, subtensor=subtensor)
"""
if lite is None:
lite = self.lite

subtensor = await self._initialize_subtensor(subtensor)

if (
Expand Down Expand Up @@ -1608,8 +1621,14 @@ async def _get_all_stakes_from_chain(self):
)
return subnet_state

self.alpha_stake = subnet_state.alpha_stake
self.tao_stake = [b * 0.018 for b in subnet_state.tao_stake]
self.alpha_stake = self._create_tensor(
[b.tao for b in subnet_state.alpha_stake],
dtype=self._dtype_registry["float32"],
)
self.tao_stake = self._create_tensor(
[b.tao * ROOT_TAO_STAKES_WEIGHT for b in subnet_state.tao_stake],
dtype=self._dtype_registry["float32"],
)
self.total_stake = self.stake = self._create_tensor(
[stake.tao for stake in subnet_state.total_stake],
dtype=self._dtype_registry["float32"],
Expand Down Expand Up @@ -1641,7 +1660,7 @@ def __init__(
def sync(
self,
block: Optional[int] = None,
lite: bool = True,
lite: Optional[bool] = None,
subtensor: Optional["Subtensor"] = None,
):
"""
Expand All @@ -1652,8 +1671,9 @@ def sync(
Args:
block (Optional[int]): A specific block number to synchronize with. If None, the metagraph syncs with the
latest block. This allows for historical analysis or specific state examination of the network.
lite (bool): If True, a lite version of the metagraph is used for quicker synchronization. This is
lite (Optional[bool]): If True, a lite version of the metagraph is used for quicker synchronization. This is
beneficial when full detail is not necessary, allowing for reduced computational and time overhead.
Defaults to `True`.
subtensor (Optional[bittensor.core.subtensor.Subtensor]): An instance of the subtensor class from Bittensor,
providing an interface to the underlying blockchain data. If provided, this instance is used for data
retrieval during synchronization.
Expand Down Expand Up @@ -1688,6 +1708,8 @@ def sync(

metagraph.sync(block=history_block, lite=False, subtensor=subtensor)
"""
if lite is None:
lite = self.lite

# Initialize subtensor
subtensor = self._initialize_subtensor(subtensor=subtensor)
Expand Down Expand Up @@ -1908,7 +1930,7 @@ def _get_all_stakes_from_chain(self):
dtype=self._dtype_registry["float32"],
)
self.tao_stake = self._create_tensor(
[b.tao * 0.018 for b in subnet_state.tao_stake],
[b.tao * ROOT_TAO_STAKES_WEIGHT for b in subnet_state.tao_stake],
dtype=self._dtype_registry["float32"],
)
self.total_stake = self.stake = self._create_tensor(
Expand Down