test(test_completion.py): fix test to use mock response - prevent hang

This commit is contained in:
Krrish Dholakia 2024-06-07 07:05:32 -07:00
parent 48e743a81a
commit a4c20294e0

View file

@ -2538,6 +2538,7 @@ def test_replicate_custom_prompt_dict():
"content": "what is yc write 1 paragraph",
}
],
mock_response="hello world",
repetition_penalty=0.1,
num_retries=3,
)