Skip to content

Commit f4973f7

Browse files
Update tests/conftest.py
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> Signed-off-by: Patrick von Platen <patrick.v.platen@gmail.com>
1 parent 9490ecb commit f4973f7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/conftest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -829,7 +829,7 @@ def get_inputs(
829829
text_prompt_kwargs = {"multi_modal_data": multi_modal_data or None}
830830
if isinstance(prompt, str):
831831
text_prompt_kwargs["prompt"] = prompt
832-
elif isinstance(prompt, list) and isinstance(prompt[0], int):
832+
elif isinstance(prompt, list):
833833
text_prompt_kwargs["prompt_token_ids"] = prompt
834834
else:
835835
text_prompt_kwargs["prompt_embeds"] = prompt

0 commit comments

Comments
 (0)