style: fix linting errors

This commit is contained in:
Krrish Dholakia 2023-10-16 17:35:08 -07:00
parent 5ef054f105
commit 7572086231
3 changed files with 7 additions and 7 deletions

View file

@ -18,7 +18,7 @@ class VLLMError(Exception):
# check if vllm is installed
def validate_environment(model: str, llm: Any =None):
try:
from vllm import LLM, SamplingParams
from vllm import LLM, SamplingParams # type: ignore
if llm is None:
llm = LLM(model=model)
return llm, SamplingParams