Merge pull request #2870 from opentensor/fix/roman/selective-metagraph

Selective-metagraph -> MetagraphInfo

Author: basfroman

bittensor/core/async_subtensor.py

@@ -23,7 +23,6 @@
     NeuronInfo,
     ProposalVoteData,
     StakeInfo,
-    SelectiveMetagraphIndex,
     SubnetHyperparameters,
     SubnetIdentity,
     SubnetInfo,
@@ -1543,59 +1542,32 @@ async def get_minimum_required_stake(self):
     async def get_metagraph_info(
         self,
         netuid: int,
-        field_indices: Optional[list["SelectiveMetagraphIndex"]] = None,
         block: Optional[int] = None,
         block_hash: Optional[str] = None,
         reuse_block: bool = False,
     ) -> Optional[MetagraphInfo]:
         """
-        Retrieves full or partial metagraph information for the specified subnet (netuid).
+        Retrieves the MetagraphInfo dataclass from the node for a single subnet (netuid)
 
         Arguments:
-            netuid: The NetUID of the subnet to query.
-            field_indices: An optional list of SelectiveMetagraphIndex values specifying which fields to retrieve. If
-                not provided, all available fields will be returned.
+            netuid: The NetUID of the subnet.
             block: the block number at which to retrieve the hyperparameter. Do not specify if using block_hash or
                 reuse_block
             block_hash: The hash of blockchain block number for the query. Do not specify if using
                 block or reuse_block
             reuse_block: Whether to reuse the last-used block hash. Do not set if using block_hash or block.
 
         Returns:
-            Optional[MetagraphInfo]: A MetagraphInfo object containing the requested subnet data, or None if the subnet
-                with the given netuid does not exist.
-
-        Example:
-            meta_info = await subtensor.get_metagraph_info(netuid=2)
-
-            partial_meta_info = await subtensor.get_metagraph_info(
-                netuid=2,
-                field_indices=[SelectiveMetagraphIndex.Name, SelectiveMetagraphIndex.OwnerHotkeys]
-            )
+            MetagraphInfo dataclass
         """
-        indexes = SelectiveMetagraphIndex.all_indices()
-
-        if field_indices:
-            if isinstance(field_indices, list) and all(
-                isinstance(f, (SelectiveMetagraphIndex, int)) for f in field_indices
-            ):
-                indexes = [
-                    f.value if isinstance(f, SelectiveMetagraphIndex) else f
-                    for f in field_indices
-                ]
-            else:
-                raise ValueError(
-                    "`field_indices` must be a list of SelectiveMetagraphIndex items."
-                )
-
         block_hash = await self.determine_block_hash(block, block_hash, reuse_block)
         if not block_hash and reuse_block:
             block_hash = self.substrate.last_block_hash
 
         query = await self.substrate.runtime_call(
             "SubnetInfoRuntimeApi",
-            "get_selective_metagraph",
-            params=[netuid, indexes if 0 in indexes else [0] + indexes],
+            "get_metagraph",
+            params=[netuid],
             block_hash=block_hash,
         )
         if query.value is None:

bittensor/core/subtensor.py

@@ -21,7 +21,6 @@
     MetagraphInfo,
     NeuronInfo,
     NeuronInfoLite,
-    SelectiveMetagraphIndex,
     StakeInfo,
     SubnetInfo,
     SubnetIdentity,
@@ -1190,53 +1189,24 @@ def get_minimum_required_stake(self) -> Balance:
         return Balance.from_rao(getattr(result, "value", 0))
 
     def get_metagraph_info(
-        self,
-        netuid: int,
-        field_indices: Optional[list[Union["SelectiveMetagraphIndex", int]]] = None,
-        block: Optional[int] = None,
+        self, netuid: int, block: Optional[int] = None
     ) -> Optional[MetagraphInfo]:
         """
-        Retrieves full or partial metagraph information for the specified subnet (netuid).
+        Retrieves the MetagraphInfo dataclass from the node for a single subnet (netuid)
 
         Arguments:
-            netuid: The NetUID of the subnet to query.
-            field_indices: An optional list of SelectiveMetagraphIndex values specifying which fields to retrieve. If
-                not provided, all available fields will be returned.
-            block: The block number at which to query the data. If not specified, the current block or one determined
-                via reuse_block or block_hash will be used.
+            netuid: The NetUID of the subnet.
+            block: the block number at which to retrieve the hyperparameter. Do not specify if using block_hash or
+                reuse_block
 
         Returns:
-            Optional[MetagraphInfo]: A MetagraphInfo object containing the requested subnet data, or None if the subnet
-                with the given netuid does not exist.
-
-        Example:
-            meta_info = subtensor.get_metagraph_info(netuid=2)
-
-            partial_meta_info = subtensor.get_metagraph_info(
-                netuid=2,
-                field_indices=[SelectiveMetagraphIndex.Name, SelectiveMetagraphIndex.OwnerHotkeys]
-            )
+            MetagraphInfo dataclass
         """
-        indexes = SelectiveMetagraphIndex.all_indices()
-
-        if field_indices:
-            if isinstance(field_indices, list) and all(
-                isinstance(f, (SelectiveMetagraphIndex, int)) for f in field_indices
-            ):
-                indexes = [
-                    f.value if isinstance(f, SelectiveMetagraphIndex) else f
-                    for f in field_indices
-                ]
-            else:
-                raise ValueError(
-                    "`field_indices` must be a list of SelectiveMetagraphIndex items."
-                )
-
         block_hash = self.determine_block_hash(block)
         query = self.substrate.runtime_call(
             "SubnetInfoRuntimeApi",
-            "get_selective_metagraph",
-            params=[netuid, indexes if 0 in indexes else [0] + indexes],
+            "get_metagraph",
+            params=[netuid],
             block_hash=block_hash,
         )
         if query.value is None: