diff --git a/tests/litellm_utils_tests/test_health_check.py b/tests/litellm_utils_tests/test_health_check.py index b459c3cfc9..963fe4f5b9 100644 --- a/tests/litellm_utils_tests/test_health_check.py +++ b/tests/litellm_utils_tests/test_health_check.py @@ -497,7 +497,7 @@ async def test_perform_health_check_filters_by_model_id(): captured_list = [] - async def mock_perform_health_check(m_list, details=True): + async def mock_perform_health_check(m_list, details=True, **kwargs): captured_list.append(m_list) return [{"model": "gpt-4", "api_key": m_list[0]["litellm_params"]["api_key"]}], []