fix: fix colpali test

joein · joein · commit 7ebe90df6fb1 · 2025-02-06T14:18:29.000+01:00
diff --git a/tests/test_late_interaction_multimodal.py b/tests/test_late_interaction_multimodal.py
@@ -41,8 +41,8 @@
 queries = ["hello world", "flag embedding"]
 images = [
     TEST_MISC_DIR / "image.jpeg",
-    str(TEST_MISC_DIR / "small_image.jpeg"),
-    Image.open((TEST_MISC_DIR / "small_image.jpeg")),
+    str(TEST_MISC_DIR / "image.jpeg"),
+    Image.open((TEST_MISC_DIR / "image.jpeg")),
 ]
 
 

Original file line number	Diff line number	Diff line change
`@@ -41,8 +41,8 @@`
`41`	`41`	`queries = ["hello world", "flag embedding"]`
`42`	`42`	`images = [`
`43`	`43`	`TEST_MISC_DIR / "image.jpeg",`
`44`		`- str(TEST_MISC_DIR / "small_image.jpeg"),`
`45`		`- Image.open((TEST_MISC_DIR / "small_image.jpeg")),`
	`44`	`+ str(TEST_MISC_DIR / "image.jpeg"),`
	`45`	`+ Image.open((TEST_MISC_DIR / "image.jpeg")),`
`46`	`46`	`]`
`47`	`47`
`48`	`48`