Skip to content

Commit 7edfefd

Browse files
committed
update get_all_subnet_ema_tao_inflow
1 parent 2ae6950 commit 7edfefd

File tree

1 file changed

+10
-7
lines changed

1 file changed

+10
-7
lines changed

bittensor_cli/src/bittensor/subtensor_interface.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2231,7 +2231,7 @@ async def get_all_subnet_ema_tao_inflow(
22312231
self,
22322232
block_hash: Optional[str] = None,
22332233
page_size: int = 100,
2234-
) -> dict[int, tuple[int, float]]:
2234+
) -> dict[int, Balance]:
22352235
"""
22362236
Query EMA TAO inflow for all subnets.
22372237
@@ -2243,20 +2243,23 @@ async def get_all_subnet_ema_tao_inflow(
22432243
page_size: The page size for batch queries (default: 100).
22442244
22452245
Returns:
2246-
Dict mapping netuid -> (block_number, Balance(EMA TAO inflow)).
2246+
Dict mapping netuid -> Balance(EMA TAO inflow).
22472247
"""
22482248
query = await self.substrate.query_map(
22492249
module="SubtensorModule",
22502250
storage_function="SubnetEmaTaoFlow",
22512251
page_size=page_size,
22522252
block_hash=block_hash,
22532253
)
2254-
tao_inflow_ema = {}
2254+
ema_map = {}
22552255
async for netuid, value in query:
2256-
block_updated, _tao_ema = value
2257-
ema_value = fixed_to_float(_tao_ema)
2258-
tao_inflow_ema[netuid] = (block_updated, Balance.from_rao(ema_value))
2259-
return tao_inflow_ema
2256+
if not value:
2257+
ema_map[netuid] = Balance.from_rao(0)
2258+
else:
2259+
_, raw_ema_value = value
2260+
ema_value = fixed_to_float(raw_ema_value)
2261+
ema_map[netuid] = Balance.from_rao(ema_value)
2262+
return ema_map
22602263

22612264
async def get_subnet_ema_tao_inflow(
22622265
self,

0 commit comments

Comments
 (0)