typing

Author: Andrei Neagu

packages/service-library/src/servicelib/fastapi/docker.py

@@ -2,7 +2,7 @@
 import logging
 from collections.abc import AsyncIterator
 from contextlib import AsyncExitStack
-from typing import Any, Final
+from typing import Final
 
 import aiodocker
 import aiohttp
@@ -21,9 +21,7 @@
 _DOCKER_API_PROXY_SETTINGS: Final[str] = "docker_api_proxy_settings"
 
 
-def get_remote_docker_client_main_lifespan(
-    settings: DockerApiProxysettings,
-) -> dict[str, Any]:
+def get_remote_docker_client_main_lifespan(settings: DockerApiProxysettings) -> State:
     return {_DOCKER_API_PROXY_SETTINGS: settings}
 
 

packages/service-library/src/servicelib/fastapi/prometheus_instrumentation.py

@@ -1,7 +1,7 @@
 # pylint: disable=protected-access
 
 from collections.abc import AsyncIterator
-from typing import Any, Final
+from typing import Final
 
 from fastapi import FastAPI
 from fastapi_lifespan_manager import State
@@ -57,9 +57,7 @@ def _on_shutdown() -> None:
 _PROMETHEUS_INSTRUMENTATION_ENABLED: Final[str] = "prometheus_instrumentation_enabled"
 
 
-def get_prometheus_instrumentationmain_main_lifespan(
-    *, enabled: bool
-) -> dict[str, Any]:
+def get_prometheus_instrumentationmain_main_lifespan(*, enabled: bool) -> State:
     return {_PROMETHEUS_INSTRUMENTATION_ENABLED: enabled}