Merge branch 'staging' into fix/zyzniewski/unittests_warnings

Author: basfroman

.github/workflows/e2e-subtensor-tests.yaml

@@ -107,7 +107,7 @@ jobs:
       - name: Run tests with retry
         run: |
           set +e
-          for i in 1 2; do
+          for i in 1 2 3; do
             echo "🔁 Attempt $i: Running tests"
             uv run pytest ${{ matrix.test-file }} -s
             status=$?
@@ -116,8 +116,8 @@ jobs:
               break
             else
               echo "❌ Tests failed on attempt $i"
-              if [ $i -eq 2 ]; then
-                echo "Tests failed after 2 attempts"
+              if [ $i -eq 3 ]; then
+                echo "Tests failed after 3 attempts"
                 exit 1
               fi
               echo "Retrying..."

.github/workflows/monitor_requirements_size_master.yml

@@ -0,0 +1,88 @@
+# This workflow measures the disk size of a virtual environment
+# after installing the Bittensor SDK across multiple Python versions.
+# It runs only when a new pull request targets the master branch,
+# and posts a comment with the results.
+name: Monitor SDK Requirements Size
+
+on:
+  pull_request:
+    types: [opened]
+    branches: [master]
+
+permissions:
+  pull-requests: write
+  contents: read
+
+jobs:
+  measure-venv:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.9", "3.10", "3.11", "3.12", "3.13"]
+    outputs:
+      py39: ${{ steps.set-output.outputs.py39 }}
+      py310: ${{ steps.set-output.outputs.py310 }}
+      py311: ${{ steps.set-output.outputs.py311 }}
+      py312: ${{ steps.set-output.outputs.py312 }}
+      py313: ${{ steps.set-output.outputs.py313 }}
+
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Create virtualenv and install
+        run: |
+          python -m venv venv
+          source venv/bin/activate
+          pip install --upgrade pip
+          pip install .
+
+      - name: Measure venv size
+        id: set-output
+        run: |
+          SIZE=$(du -sm venv | cut -f1)
+          VERSION=${{ matrix.python-version }}
+          echo "Detected size: $SIZE MB for Python $VERSION"
+          case "$VERSION" in
+            3.9)  echo "py39=$SIZE" >> $GITHUB_OUTPUT ;;
+            3.10) echo "py310=$SIZE" >> $GITHUB_OUTPUT ;;
+            3.11) echo "py311=$SIZE" >> $GITHUB_OUTPUT ;;
+            3.12) echo "py312=$SIZE" >> $GITHUB_OUTPUT ;;
+            3.13) echo "py313=$SIZE" >> $GITHUB_OUTPUT ;;
+          esac
+
+  comment-on-pr:
+    if: github.event_name == 'pull_request' && github.base_ref == 'master'
+    needs: measure-venv
+    runs-on: ubuntu-latest
+    steps:
+      - name: Post venv size summary to PR
+        uses: actions/github-script@v7
+        with:
+          github-token: ${{ secrets.GITHUB_TOKEN }}
+          script: |
+            const sizes = {
+              "3.9": "${{ needs.measure-venv.outputs.py39 || 'N/A' }}",
+              "3.10": "${{ needs.measure-venv.outputs.py310 || 'N/A' }}",
+              "3.11": "${{ needs.measure-venv.outputs.py311 || 'N/A' }}",
+              "3.12": "${{ needs.measure-venv.outputs.py312 || 'N/A' }}",
+              "3.13": "${{ needs.measure-venv.outputs.py313 || 'N/A' }}",
+            };
+
+            const body = [
+              '**Bittensor SDK virtual environment sizes by Python version:**',
+              '',
+              '```'
+            ]
+              .concat(Object.entries(sizes).map(([v, s]) => `Python ${v}: ${s} MB`))
+              .concat(['```'])
+              .join('\n');
+
+            github.rest.issues.createComment({
+              issue_number: context.issue.number,
+              owner: context.repo.owner,
+              repo: context.repo.repo,
+              body
+            });

bittensor/core/async_subtensor.py

@@ -16,17 +16,18 @@
 
 from bittensor.core.chain_data import (
     DelegateInfo,
-    StakeInfo,
+    DynamicInfo,
     MetagraphInfo,
     NeuronInfoLite,
     NeuronInfo,
     ProposalVoteData,
+    StakeInfo,
+    SelectiveMetagraphIndex,
     SubnetHyperparameters,
     SubnetIdentity,
     SubnetInfo,
     WeightCommitInfo,
     decode_account_id,
-    DynamicInfo,
 )
 from bittensor.core.chain_data.chain_identity import ChainIdentity
 from bittensor.core.chain_data.delegate_info import DelegatedInfo
@@ -1470,32 +1471,56 @@ 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 the MetagraphInfo dataclass from the node for a single subnet (netuid)
+        Retrieves full or partial metagraph information for the specified subnet (netuid).
 
         Arguments:
-            netuid: The NetUID of the subnet.
+            netuid (int): The NetUID of the subnet to query.
+            field_indices (Optional[list[SelectiveMetagraphIndex]]): 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 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:
-            MetagraphInfo dataclass
+            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]
+            )
         """
+        indexes = SelectiveMetagraphIndex.all_indices()
+
+        if field_indices:
+            if isinstance(field_indices, list) and all(
+                isinstance(f, SelectiveMetagraphIndex) for f in field_indices
+            ):
+                indexes = [f.value 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_metagraph",
-            params=[netuid],
+            "get_selective_metagraph",
+            params=[netuid, indexes if 0 in indexes else [0] + indexes],
             block_hash=block_hash,
         )
         if query.value is None:
@@ -4177,7 +4202,7 @@ async def unstake(
                 removed.
             hotkey_ss58 (Optional[str]): The ``SS58`` address of the hotkey account to unstake from.
             netuid (Optional[int]): The unique identifier of the subnet.
-            amount (Balance): The amount of TAO to unstake. If not specified, unstakes all.
+            amount (Balance): The amount of alpha to unstake. If not specified, unstakes all.
             wait_for_inclusion (bool): Waits for the transaction to be included in a block.
             wait_for_finalization (bool): Waits for the transaction to be finalized on the blockchain.
             safe_staking (bool): If true, enables price safety checks to protect against fluctuating prices. The unstake

bittensor/core/chain_data/__init__.py

@@ -16,6 +16,7 @@
     MetagraphInfoEmissions,
     MetagraphInfoPool,
     MetagraphInfoParams,
+    SelectiveMetagraphIndex,
 )
 from .neuron_info import NeuronInfo
 from .neuron_info_lite import NeuronInfoLite
@@ -50,6 +51,7 @@
     ProposalCallData,
     ProposalVoteData,
     ScheduledColdkeySwapInfo,
+    SelectiveMetagraphIndex,
     StakeInfo,
     SubnetHyperparameters,
     SubnetIdentity,