fix predibase mock test

2024-07-31 08:16:24 -07:00 · 2024-07-31 08:16:24 -07:00 · 7db85fbdb7
commit 7db85fbdb7
parent c583cbdbcd
1 changed files with 1 additions and 1 deletions
--- 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"}