From 7445b5adc6be3e3d53b48bda6b714368389a0973 Mon Sep 17 00:00:00 2001 From: ibraheem-opentensor Date: Tue, 18 Feb 2025 17:43:24 -0800 Subject: [PATCH 1/2] Adds stake weight to metagraph info --- bittensor/core/chain_data/metagraph_info.py | 3 ++- bittensor/core/metagraph.py | 5 ++--- bittensor/core/settings.py | 1 + 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/bittensor/core/chain_data/metagraph_info.py b/bittensor/core/chain_data/metagraph_info.py index f79f0d4cc4..6700b11ddd 100644 --- a/bittensor/core/chain_data/metagraph_info.py +++ b/bittensor/core/chain_data/metagraph_info.py @@ -1,6 +1,7 @@ from dataclasses import dataclass from typing import Optional, Union +from bittensor.core import settings from bittensor.core.chain_data.axon_info import AxonInfo from bittensor.core.chain_data.chain_identity import ChainIdentity from bittensor.core.chain_data.info_base import InfoBase @@ -234,7 +235,7 @@ def _from_dict(cls, decoded: dict) -> "MetagraphInfo": rank=[u16tf(rk) for rk in decoded.get("rank", [])], block_at_registration=decoded["block_at_registration"], alpha_stake=[_tbwu(ast, _netuid) for ast in decoded["alpha_stake"]], - tao_stake=[_tbwu(ts) for ts in decoded["tao_stake"]], + tao_stake=[_tbwu(ts)*settings.ROOT_TAO_STAKE_WEIGHT for ts in decoded["tao_stake"]], total_stake=[_tbwu(ts, _netuid) for ts in decoded["total_stake"]], # Dividend break down tao_dividends_per_hotkey=[ diff --git a/bittensor/core/metagraph.py b/bittensor/core/metagraph.py index 2f68709ab2..acf44c7c70 100644 --- a/bittensor/core/metagraph.py +++ b/bittensor/core/metagraph.py @@ -42,7 +42,6 @@ Tensor = Union["torch.nn.Parameter", NDArray] -ROOT_TAO_STAKES_WEIGHT = 0.18 METAGRAPH_STATE_DICT_NDARRAY_KEYS = [ @@ -1598,7 +1597,7 @@ async def _get_all_stakes_from_chain(self): dtype=self._dtype_registry["float32"], ) self.tao_stake = self._create_tensor( - [b.tao * ROOT_TAO_STAKES_WEIGHT for b in subnet_state.tao_stake], + [b.tao * settings.ROOT_TAO_STAKE_WEIGHT for b in subnet_state.tao_stake], dtype=self._dtype_registry["float32"], ) self.total_stake = self.stake = self._create_tensor( @@ -1902,7 +1901,7 @@ def _get_all_stakes_from_chain(self): dtype=self._dtype_registry["float32"], ) self.tao_stake = self._create_tensor( - [b.tao * ROOT_TAO_STAKES_WEIGHT for b in subnet_state.tao_stake], + [b.tao * settings.ROOT_TAO_STAKE_WEIGHT for b in subnet_state.tao_stake], dtype=self._dtype_registry["float32"], ) self.total_stake = self.stake = self._create_tensor( diff --git a/bittensor/core/settings.py b/bittensor/core/settings.py index ec51adafdc..b33d174986 100644 --- a/bittensor/core/settings.py +++ b/bittensor/core/settings.py @@ -6,6 +6,7 @@ from munch import munchify +ROOT_TAO_STAKE_WEIGHT = 0.18 READ_ONLY = os.getenv("READ_ONLY") == "1" From b7eec15bb3e77e70939cc09b017ce7e91e511763 Mon Sep 17 00:00:00 2001 From: ibraheem-opentensor Date: Tue, 18 Feb 2025 17:43:57 -0800 Subject: [PATCH 2/2] ruff --- bittensor/core/chain_data/metagraph_info.py | 5 ++++- bittensor/core/metagraph.py | 10 ++++++++-- 2 files changed, 12 insertions(+), 3 deletions(-) diff --git a/bittensor/core/chain_data/metagraph_info.py b/bittensor/core/chain_data/metagraph_info.py index 6700b11ddd..1ed2733124 100644 --- a/bittensor/core/chain_data/metagraph_info.py +++ b/bittensor/core/chain_data/metagraph_info.py @@ -235,7 +235,10 @@ def _from_dict(cls, decoded: dict) -> "MetagraphInfo": rank=[u16tf(rk) for rk in decoded.get("rank", [])], block_at_registration=decoded["block_at_registration"], alpha_stake=[_tbwu(ast, _netuid) for ast in decoded["alpha_stake"]], - tao_stake=[_tbwu(ts)*settings.ROOT_TAO_STAKE_WEIGHT for ts in decoded["tao_stake"]], + tao_stake=[ + _tbwu(ts) * settings.ROOT_TAO_STAKE_WEIGHT + for ts in decoded["tao_stake"] + ], total_stake=[_tbwu(ts, _netuid) for ts in decoded["total_stake"]], # Dividend break down tao_dividends_per_hotkey=[ diff --git a/bittensor/core/metagraph.py b/bittensor/core/metagraph.py index acf44c7c70..06b9a66bb2 100644 --- a/bittensor/core/metagraph.py +++ b/bittensor/core/metagraph.py @@ -1597,7 +1597,10 @@ async def _get_all_stakes_from_chain(self): dtype=self._dtype_registry["float32"], ) self.tao_stake = self._create_tensor( - [b.tao * settings.ROOT_TAO_STAKE_WEIGHT for b in subnet_state.tao_stake], + [ + b.tao * settings.ROOT_TAO_STAKE_WEIGHT + for b in subnet_state.tao_stake + ], dtype=self._dtype_registry["float32"], ) self.total_stake = self.stake = self._create_tensor( @@ -1901,7 +1904,10 @@ def _get_all_stakes_from_chain(self): dtype=self._dtype_registry["float32"], ) self.tao_stake = self._create_tensor( - [b.tao * settings.ROOT_TAO_STAKE_WEIGHT for b in subnet_state.tao_stake], + [ + b.tao * settings.ROOT_TAO_STAKE_WEIGHT + for b in subnet_state.tao_stake + ], dtype=self._dtype_registry["float32"], ) self.total_stake = self.stake = self._create_tensor(