Skip to content

Commit

Permalink
🎨 fix ruff errors
Browse files Browse the repository at this point in the history
Signed-off-by: Prashant Gupta <[email protected]>
  • Loading branch information
prashantgupta24 committed Jun 19, 2024
1 parent 1cae820 commit 52f4793
Showing 1 changed file with 9 additions and 8 deletions.
17 changes: 9 additions & 8 deletions src/vllm_tgis_adapter/__main__.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,13 +40,14 @@
import argparse
from collections.abc import AsyncGenerator

from fastapi import FastAPI, Request
from vllm.config import ModelConfig
from vllm.entrypoints.openai.protocol import (
ChatCompletionRequest,
CompletionRequest,
EmbeddingRequest,
)

from fastapi import FastAPI, Request # noqa: TCH002
from vllm.entrypoints.openai.protocol import (
ChatCompletionRequest, # noqa: TCH002
CompletionRequest, # noqa: TCH002
EmbeddingRequest, # noqa: TCH002
)

TIMEOUT_KEEP_ALIVE = 5 # seconds

Expand Down Expand Up @@ -314,14 +315,14 @@ async def _force_log() -> None:
if event_loop is None:
event_loop = asyncio.new_event_loop()

event_loop.create_task(
event_loop.create_task( # noqa: RUF006
run_http_server(
args,
engine,
model_config,
)
)
grpc = event_loop.create_task(
grpc = event_loop.create_task( # noqa: RUF006
grpc_server(
engine,
disable_log_stats=engine_args.disable_log_stats,
Expand Down

0 comments on commit 52f4793

Please sign in to comment.