Skip to content

Commit e6743fb

Browse files
committed
Init
1 parent 2eab16c commit e6743fb

File tree

3 files changed

+29
-0
lines changed

3 files changed

+29
-0
lines changed

bittensor/core/async_subtensor.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
from scalecodec.type_registry import load_type_registry_preset
1616

1717
from bittensor.core.types import SubtensorMixin
18+
from bittensor.core.subtensor_plugins.subtensor_module import SubtensorModule
1819
from bittensor.core.chain_data import (
1920
DelegateInfo,
2021
StakeInfo,
@@ -131,6 +132,7 @@ def __init__(
131132
logging.info(
132133
f"Connected to {self.network} network and {self.chain_endpoint}."
133134
)
135+
self.subtensor_module = SubtensorModule(self)
134136

135137
async def close(self):
136138
"""Close the connection."""

bittensor/core/subtensor.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
from scalecodec.type_registry import load_type_registry_preset
1414

1515
from bittensor.core.types import SubtensorMixin
16+
from bittensor.core.subtensor_plugins.subtensor_module import SubtensorModule
1617
from bittensor.core.chain_data import (
1718
custom_rpc_type_registry,
1819
decode_account_id,
@@ -130,6 +131,7 @@ def __init__(
130131
logging.info(
131132
f"Connected to {self.network} network and {self.chain_endpoint}."
132133
)
134+
self.subtensor_module = SubtensorModule(self)
133135

134136
def close(self):
135137
"""
Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
2+
from typing import Union, TYPE_CHECKING
3+
4+
5+
if TYPE_CHECKING:
6+
from bittensor.core.subtensor import Subtensor
7+
from bittensor.core.async_subtensor import AsyncSubtensor
8+
9+
10+
class SubtensorModule:
11+
def __init__(self, subtensor: Union["Subtensor", "AsyncSubtensor"]):
12+
self._subtensor = subtensor
13+
14+
def tempo(self, netuid: int):
15+
return self._subtensor.substrate.query(
16+
module="SubtensorModule",
17+
storage_function="Tempo",
18+
params=[netuid]
19+
)
20+
21+
def tempos(self):
22+
return self._subtensor.substrate.query_map(
23+
module="SubtensorModule",
24+
storage_function="Tempo",
25+
)

0 commit comments

Comments
 (0)