Merge remote-tracking branch 'origin/fix/thewhaleking/add-all-easy-imports' into fix/thewhaleking/add-all-easy-imports

Author: thewhaleking

bittensor/core/async_subtensor.py

@@ -303,7 +303,11 @@ def _get_substrate(
             return RetryAsyncSubstrate(
                 url=self.chain_endpoint,
                 fallback_chains=fallback_chains,
+                ss58_format=SS58_FORMAT,
+                type_registry=TYPE_REGISTRY,
                 retry_forever=retry_forever,
+                use_remote_preset=True,
+                chain_name="Bittensor",
                 _mock=_mock,
             )
         return AsyncSubstrateInterface(

bittensor/core/subtensor.py

@@ -185,6 +185,10 @@ def _get_substrate(
         if fallback_chains or retry_forever:
             return RetrySyncSubstrate(
                 url=self.chain_endpoint,
+                ss58_format=SS58_FORMAT,
+                type_registry=TYPE_REGISTRY,
+                use_remote_preset=True,
+                chain_name="Bittensor",
                 fallback_chains=fallback_chains,
                 retry_forever=retry_forever,
                 _mock=_mock,

bittensor/core/subtensor_api/__init__.py

@@ -22,7 +22,7 @@ class SubtensorApi:
     """Subtensor API class.
 
     Arguments:
-        network: The network to connect to. Defaults to `None` -> `finney`.
+        network: The network to connect to. Defaults to `None` -> "finney".
         config: Bittensor configuration object. Defaults to `None`.
         legacy_methods: If `True`, all methods from the Subtensor class will be added to the root level of this class.
         fallback_chains (list): List of fallback chains to use if no network is specified. Defaults to `None`.
@@ -144,15 +144,31 @@ def __repr__(self):
         return self.__str__()
 
     def __enter__(self):
+        if self.is_async:
+            raise NotImplementedError(
+                "Async version of SubtensorApi cannot be used with sync context manager."
+            )
         return self
 
     def __exit__(self, exc_type, exc_val, exc_tb):
+        if self.is_async:
+            raise NotImplementedError(
+                "Async version of SubtensorApi cannot be used with sync context manager."
+            )
         self.close()
 
     async def __aenter__(self):
+        if not self.is_async:
+            raise NotImplementedError(
+                "Sync version of SubtensorApi cannot be used with async context manager."
+            )
         return await self._subtensor.__aenter__()
 
     async def __aexit__(self, exc_type, exc_val, exc_tb):
+        if not self.is_async:
+            raise NotImplementedError(
+                "Sync version of SubtensorApi cannot be used with async context manager."
+            )
         await self.substrate.close()
 
     @classmethod

pyproject.toml

@@ -36,7 +36,7 @@ dependencies = [
     "uvicorn",
     "bittensor-drand>=0.5.0",
     "bittensor-wallet>=3.0.8",
-    "async-substrate-interface>=1.1.0"
+    "async-substrate-interface>=1.2.0"
 ]
 
 [project.optional-dependencies]