Skip to content

GitLab

Unverified Commit 5446ad1d authored by Yannick Schnider's avatar Yannick Schnider Committed by GitHub
Browse files

[test utils] correct wrong typing (#26159) 


Signed-off-by: default avatarYannick Schnider <yannick.schnider1@ibm.com>
parent f9a8084e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/conftest.py
View file @ 5446ad1d
......@@ -756,7 +756,7 @@ class VllmRunner:
def get_inputs(
self,
prompts: Union[list[str], list[torch.Tensor], list[int]],
prompts: Union[list[str], list[torch.Tensor], list[list[int]]],
images: Optional[PromptImageInput] = None,
videos: Optional[PromptVideoInput] = None,
audios: Optional[PromptAudioInput] = None,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment