Refactored the logic of __aenter__ for AsyncSubtensor so that it doesn't close the websocket immediately.

Author: thewhaleking

bittensor/core/async_subtensor.py

@@ -173,8 +173,8 @@ async def __aenter__(self):
             f"[magenta]Connecting to Substrate:[/magenta] [blue]{self}[/blue][magenta]...[/magenta]"
         )
         try:
-            async with self.substrate:
-                return self
+            await self.substrate.initialize()
+            return self
         except TimeoutError:
             logging.error(
                 f"[red]Error[/red]: Timeout occurred connecting to substrate."