fix test case

landing-ai · Apr 17, 2024 · 7ed32af · 7ed32af
1 parent 56fa5d6
commit 7ed32af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_lmm.py b/tests/test_lmm.py
@@ -27,7 +27,7 @@ def create_temp_image(image_format="jpeg"):
 def test_generate_with_mock(openai_lmm_mock): # noqa: F811
  temp_image = create_temp_image()
  lmm = OpenAILMM()
- response = lmm.generate("test prompt", image=temp_image)
+ response = lmm.generate("test prompt", images=[temp_image])
  assert response == "mocked response"
  assert (
  "image_url"