test(test_completion.py): fix replicate test which was hanging
This commit is contained in:
parent
09a471da10
commit
ddc82774b0
@ -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,
|
||||
)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user