diff --git a/tests/test_litellm/proxy/config_resolvers/test_settings_store.py b/tests/test_litellm/proxy/config_resolvers/test_settings_store.py index ab1bff67c42..806b2d5e5aa 100644 --- a/tests/test_litellm/proxy/config_resolvers/test_settings_store.py +++ b/tests/test_litellm/proxy/config_resolvers/test_settings_store.py @@ -427,7 +427,7 @@ def test_settings_store_truthiness_stops_at_the_first_key() -> None: resolutions: Final[list[str]] = [] original: Final = SettingsStore._resolution_for - def counted(self: SettingsStore, key: str): # type: ignore[no-untyped-def] + def counted(self: SettingsStore, key: str): resolutions.append(key) return original(self, key) diff --git a/tests/test_litellm/proxy/management_endpoints/test_coordination_redis_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_coordination_redis_endpoints.py index faa8b851db4..7c6e8154107 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_coordination_redis_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_coordination_redis_endpoints.py @@ -623,7 +623,7 @@ def _real_proxy_config(file_general_settings: dict) -> "object": proxy_config = ProxyConfig() proxy_config._load_yaml_settings_stores({"general_settings": file_general_settings}) - proxy_config.get_config_state = MagicMock( # type: ignore[method-assign] + proxy_config.get_config_state = MagicMock( return_value={"general_settings": file_general_settings} ) return proxy_config