fix typing errors

Author: khvn26

src/common/core/utils.py

@@ -201,6 +201,8 @@ def using_database_replica(
     # Already has the desired behavior; re-export for uniform imports.
     TemporaryDirectory = tempfile.TemporaryDirectory
 else:
+    from os import PathLike
+    from types import TracebackType
 
     class TemporaryDirectory(tempfile.TemporaryDirectory):
         """
@@ -217,20 +219,34 @@ class TemporaryDirectory(tempfile.TemporaryDirectory):
 
         def __init__(
             self,
-            *args: object,
+            suffix: str | None = None,
+            prefix: str | None = None,
+            dir: str | PathLike[str] | None = None,
+            ignore_cleanup_errors: bool = False,
+            *,
             delete: bool = True,
-        ):
+        ) -> None:
             self.delete = delete
-            super().__init__(*args)
-
-        def __exit__(self, exc_type, exc, tb):
+            super().__init__(
+                suffix=suffix,
+                prefix=prefix,
+                dir=dir,
+                ignore_cleanup_errors=ignore_cleanup_errors,
+            )
+
+        def __exit__(
+            self,
+            exc_type: type[BaseException] | None,
+            exc: BaseException | None,
+            tb: TracebackType | None,
+        ) -> bool:
             """On context exit, only cleanup if delete=True."""
             if self.delete:
                 super().__exit__(exc_type, exc, tb)
             # Return False to propagate exceptions like the stdlib does.
             return False
 
-        def _cleanup(self, name, warn_message):
+        def _cleanup(self, name: str, warn_message: str) -> None:
             """
             Called by the weakref finalizer on GC in 3.11.
             Respect delete=False by doing nothing in that case.