Convert formatting to use ruff instead of yapf + isort (#26247)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
@@ -24,16 +24,13 @@ async def test_shutdown_on_engine_failure():
|
||||
|
||||
with RemoteOpenAIServer(MODEL_NAME, args) as remote_server:
|
||||
async with remote_server.get_async_client() as client:
|
||||
|
||||
with pytest.raises(
|
||||
(openai.APIConnectionError, openai.InternalServerError)):
|
||||
with pytest.raises((openai.APIConnectionError, openai.InternalServerError)):
|
||||
# Asking for lots of prompt logprobs will currently crash the
|
||||
# engine. This may change in the future when that bug is fixed
|
||||
prompt = "Hello " * 4000
|
||||
await client.completions.create(
|
||||
model=MODEL_NAME,
|
||||
prompt=prompt,
|
||||
extra_body={"prompt_logprobs": 10})
|
||||
model=MODEL_NAME, prompt=prompt, extra_body={"prompt_logprobs": 10}
|
||||
)
|
||||
|
||||
# Now the server should shut down
|
||||
return_code = remote_server.proc.wait(timeout=8)
|
||||
|
||||
Reference in New Issue
Block a user