style: apply ruff formatting and type annotation modernization

Author: future-xy

pylet/aio/__init__.py

@@ -13,7 +13,7 @@ async def main():
 """
 
 import shlex
-from typing import Dict, List, Optional, Union
+from typing import Any, Optional, Union
 
 import httpx
 
@@ -28,17 +28,19 @@ async def main():
 from pylet.aio._instance import Instance
 from pylet.errors import NotFoundError, NotInitializedError
 
-# Re-export Instance for type hints
+# Re-export for public API
 __all__ = [
+    "Instance",
+    "NotFoundError",
+    "NotInitializedError",
+    "WorkerInfo",
+    "get",
     "init",
-    "shutdown",
+    "instances",
     "is_initialized",
+    "shutdown",
     "submit",
-    "get",
-    "instances",
     "workers",
-    "Instance",
-    "WorkerInfo",
 ]
 
 
@@ -91,18 +93,17 @@ def is_initialized() -> bool:
 
 
 async def submit(
-    command: Union[str, List[str]],
+    command: Union[str, list[str]],
     *,
     name: Optional[str] = None,
     gpu: int = 0,
     cpu: int = 1,
     memory: int = 512,
-    # SLLM support parameters
     target_worker: Optional[str] = None,
-    gpu_indices: Optional[List[int]] = None,
+    gpu_indices: Optional[list[int]] = None,
     exclusive: bool = True,
-    labels: Optional[Dict[str, str]] = None,
-    env: Optional[Dict[str, str]] = None,
+    labels: Optional[dict[str, str]] = None,
+    env: Optional[dict[str, str]] = None,
     # Venv support
     venv: Optional[str] = None,
 ) -> Instance:
@@ -137,7 +138,7 @@ async def submit(
         command = shlex.join(command)
 
     # Build submission request
-    submission = {
+    submission: dict[str, Any] = {
         "command": command,
         "resource_requirements": {
             "cpu_cores": cpu,
@@ -213,7 +214,7 @@ async def get(
 async def instances(
     *,
     status: Optional[str] = None,
-) -> List[Instance]:
+) -> list[Instance]:
     """
     List all instances.
 
@@ -239,7 +240,7 @@ async def instances(
     return [Instance(data) for data in response.json()]
 
 
-async def workers() -> List[WorkerInfo]:
+async def workers() -> list[WorkerInfo]:
     """
     List all registered workers.
 

pylet/aio/_instance.py

@@ -5,7 +5,7 @@
 """
 
 import asyncio
-from typing import Any, Dict, Optional
+from typing import Any, Optional
 
 from pylet._state import _get_async_client, _get_head_address
 from pylet.errors import (
@@ -14,7 +14,6 @@
     TimeoutError,
 )
 
-
 # Terminal states
 TERMINAL_STATES = frozenset({"COMPLETED", "FAILED", "CANCELLED"})
 
@@ -27,7 +26,7 @@ class Instance:
     Wait methods automatically refresh.
     """
 
-    def __init__(self, data: Dict[str, Any]):
+    def __init__(self, data: dict[str, Any]):
         """
         Initialize instance from server data.
 
@@ -97,7 +96,7 @@ async def wait_running(self, timeout: float = 300) -> None:
             if self.status in TERMINAL_STATES:
                 raise InstanceFailedError(
                     f"Instance entered {self.status} state while waiting for RUNNING",
-                    instance=self,
+                    instance=self,  # type: ignore[arg-type]
                 )
 
             await asyncio.sleep(min(poll_interval, timeout - elapsed))