Skip to content

Commit 01965c9

Browse files
committed
fix: lowercase models to cache
1 parent 8f6f057 commit 01965c9

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

tests/test_late_interaction_multimodal.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,8 @@
6767
Image.open((TEST_MISC_DIR / "image.jpeg")),
6868
]
6969

70-
MODELS_TO_CACHE = ("Qdrant/colmodernvbert",)
70+
_MODELS_TO_CACHE = ("Qdrant/colmodernvbert",)
71+
MODELS_TO_CACHE = (model_name.lower() for model_name in _MODELS_TO_CACHE)
7172

7273

7374
@pytest.fixture(scope="module")
@@ -90,7 +91,7 @@ def get_model(model_name: str):
9091
yield get_model
9192

9293
if is_ci:
93-
for name, model in cache.items():
94+
for _, model in cache.items():
9495
delete_model_cache(model.model._model_dir)
9596
cache.clear()
9697

0 commit comments

Comments
 (0)