🔧 fix typing

Author: shroominic

codeboxapi/box/basebox.py

@@ -6,8 +6,6 @@
 from typing import List, Optional
 from uuid import UUID
 
-from typing_extensions import Self
-
 from codeboxapi.schema import CodeBoxFile, CodeBoxOutput, CodeBoxStatus
 
 
@@ -99,11 +97,11 @@ def stop(self) -> CodeBoxStatus:
     async def astop(self) -> CodeBoxStatus:
         """Async Terminate the CodeBox instance"""
 
-    def __enter__(self) -> Self:
+    def __enter__(self) -> "BaseBox":
         self.start()
         return self
 
-    async def __aenter__(self) -> Self:
+    async def __aenter__(self) -> "BaseBox":
         await self.astart()
         return self
 

codeboxapi/box/localbox.py

@@ -18,7 +18,6 @@
 
 import aiohttp
 import requests
-from typing_extensions import Self
 from websockets.client import WebSocketClientProtocol
 from websockets.client import connect as ws_connect
 from websockets.exceptions import ConnectionClosedError
@@ -37,7 +36,7 @@ class LocalBox(BaseBox):
     This is useful for testing and development.
     """
 
-    _instance: Optional[Self] = None
+    _instance: Optional["LocalBox"] = None
 
     def __new__(cls, *args, **kwargs):
         if not cls._instance: