We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8f6f057 commit 01965c9Copy full SHA for 01965c9
1 file changed
tests/test_late_interaction_multimodal.py
@@ -67,7 +67,8 @@
67
Image.open((TEST_MISC_DIR / "image.jpeg")),
68
]
69
70
-MODELS_TO_CACHE = ("Qdrant/colmodernvbert",)
+_MODELS_TO_CACHE = ("Qdrant/colmodernvbert",)
71
+MODELS_TO_CACHE = (model_name.lower() for model_name in _MODELS_TO_CACHE)
72
73
74
@pytest.fixture(scope="module")
@@ -90,7 +91,7 @@ def get_model(model_name: str):
90
91
yield get_model
92
93
if is_ci:
- for name, model in cache.items():
94
+ for _, model in cache.items():
95
delete_model_cache(model.model._model_dir)
96
cache.clear()
97
0 commit comments