fix: mypy

Author: biswapanda

deploy/sdk/src/dynamo/sdk/cli/circus.py

@@ -104,7 +104,7 @@ def get_env_or_reserved_port(env_var):
     if port_env:
         return int(port_env)
     else:
-        with reserve_free_port() as port:
+        with reserve_free_port() as port:  # type: ignore
             return port
 
 

deploy/sdk/src/dynamo/sdk/cli/utils.py

@@ -24,7 +24,7 @@
 import os
 import pathlib
 import socket
-from typing import Any, DefaultDict, Dict, Protocol, TextIO, Union
+from typing import Any, DefaultDict, Dict, Iterator, Protocol, TextIO, Union
 
 import typer
 import yaml
@@ -61,8 +61,8 @@ def dynamo_address(self) -> tuple[str, str]:
 class PortReserver:
     def __init__(self, host: str = "localhost"):
         self.host = host
-        self.socket = None
-        self.port = None
+        self.socket: socket.socket | None = None
+        self.port: int | None = None
 
     def __enter__(self) -> int:
         try:
@@ -91,7 +91,7 @@ def close_socket(self):
 @contextlib.contextmanager
 def reserve_free_port(
     host: str = "localhost",
-) -> contextlib.AbstractContextManager[int]:
+) -> Iterator[int]:
     """
     Detect free port and reserve until exit the context.
     Returns a context manager that yields the reserved port.