fix predibase mock test

This commit is contained in:
Ishaan Jaff
2024-07-31 08:16:24 -07:00
parent c583cbdbcd
commit 7db85fbdb7
+1 -1
View File
@@ -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"}