From 7db85fbdb7458875f0f08e58449f3ea9dce02c16 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 31 Jul 2024 08:16:24 -0700 Subject: [PATCH] fix predibase mock test --- litellm/tests/test_completion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index aa70de7b8b..aa37d0e11c 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -211,7 +211,7 @@ async def test_completion_databricks(sync_mode): response_format_tests(response=response) -def predibase_mock_post(url, data=None, json=None, headers=None): +def predibase_mock_post(url, data=None, json=None, headers=None, timeout=None): mock_response = MagicMock() mock_response.status_code = 200 mock_response.headers = {"Content-Type": "application/json"}