Merge pull request #1731 from opentensor/feature/gus/51818215/expand-…

…type-checking-subtensor Expands type checking to subtensor
opentensor · Mar 6, 2024 · 0c05882 · 0c05882
2 parents 106bf4a + 591df7b
commit 0c05882
Show file tree

Hide file tree

Showing 9 changed files with 320 additions and 212 deletions.
diff --git a/bittensor/__init__.py b/bittensor/__init__.py
@@ -30,7 +30,7 @@
 __version__ = "6.8.2"
 
 version_split = __version__.split(".")
-__version_as_int__ = (
+__version_as_int__: int = (
     (100 * int(version_split[0]))
     + (10 * int(version_split[1]))
     + (1 * int(version_split[2]))

diff --git a/bittensor/extrinsics/delegation.py b/bittensor/extrinsics/delegation.py
@@ -97,7 +97,7 @@ def delegate_extrinsic(
     subtensor: "bittensor.subtensor",
     wallet: "bittensor.wallet",
     delegate_ss58: Optional[str] = None,
-    amount: Union[Balance, float] = None,
+    amount: Optional[Union[Balance, float]] = None,
     wait_for_inclusion: bool = True,
     wait_for_finalization: bool = False,
     prompt: bool = False,
@@ -232,7 +232,7 @@ def undelegate_extrinsic(
     subtensor: "bittensor.subtensor",
     wallet: "bittensor.wallet",
     delegate_ss58: Optional[str] = None,
-    amount: Union[Balance, float] = None,
+    amount: Optional[Union[Balance, float]] = None,
     wait_for_inclusion: bool = True,
     wait_for_finalization: bool = False,
     prompt: bool = False,

diff --git a/bittensor/extrinsics/serving.py b/bittensor/extrinsics/serving.py
@@ -206,7 +206,7 @@ def publish_metadata(
     wallet: "bittensor.wallet",
     netuid: int,
     type: str,
-    data: str,
+    data: bytes,
     wait_for_inclusion: bool = False,
     wait_for_finalization: bool = True,
 ) -> bool:

diff --git a/bittensor/extrinsics/staking.py b/bittensor/extrinsics/staking.py
@@ -27,7 +27,7 @@ def add_stake_extrinsic(
     subtensor: "bittensor.subtensor",
     wallet: "bittensor.wallet",
     hotkey_ss58: Optional[str] = None,
-    amount: Union[Balance, float] = None,
+    amount: Optional[Union[Balance, float]] = None,
     wait_for_inclusion: bool = True,
     wait_for_finalization: bool = False,
     prompt: bool = False,
@@ -204,7 +204,7 @@ def add_stake_multiple_extrinsic(
     subtensor: "bittensor.subtensor",
     wallet: "bittensor.wallet",
     hotkey_ss58s: List[str],
-    amounts: List[Union[Balance, float]] = None,
+    amounts: Optional[List[Union[Balance, float]]] = None,
     wait_for_inclusion: bool = True,
     wait_for_finalization: bool = False,
     prompt: bool = False,

diff --git a/bittensor/extrinsics/unstaking.py b/bittensor/extrinsics/unstaking.py
@@ -75,7 +75,7 @@ def unstake_extrinsic(
     subtensor: "bittensor.subtensor",
     wallet: "bittensor.wallet",
     hotkey_ss58: Optional[str] = None,
-    amount: Union[Balance, float] = None,
+    amount: Optional[Union[Balance, float]] = None,
     wait_for_inclusion: bool = True,
     wait_for_finalization: bool = False,
     prompt: bool = False,
@@ -210,7 +210,7 @@ def unstake_multiple_extrinsic(
     subtensor: "bittensor.subtensor",
     wallet: "bittensor.wallet",
     hotkey_ss58s: List[str],
-    amounts: List[Union[Balance, float]] = None,
+    amounts: Optional[List[Union[Balance, float]]] = None,
     wait_for_inclusion: bool = True,
     wait_for_finalization: bool = False,
     prompt: bool = False,

diff --git a/bittensor/metagraph.py b/bittensor/metagraph.py
@@ -748,7 +748,7 @@ def _process_root_weights(
 
         """
         data_array = []
-        n_subnets = subtensor.get_total_subnets()
+        n_subnets = subtensor.get_total_subnets() or 0
         subnets = subtensor.get_subnets()
         for item in data:
             if len(item) == 0: