diff --git a/src/vidore_benchmark/__init__.py b/src/vidore_benchmark/__init__.py index ae48564b..fdae5877 100644 --- a/src/vidore_benchmark/__init__.py +++ b/src/vidore_benchmark/__init__.py @@ -7,6 +7,7 @@ CohereAPIRetriever, ColIdefics3Retriever, ColPaliRetriever, + ColQwen2_5_Retriever, ColQwen2Retriever, DSEQwen2Retriever, DummyVisionRetriever, diff --git a/src/vidore_benchmark/retrievers/__init__.py b/src/vidore_benchmark/retrievers/__init__.py index af15b21f..3969189d 100644 --- a/src/vidore_benchmark/retrievers/__init__.py +++ b/src/vidore_benchmark/retrievers/__init__.py @@ -6,8 +6,8 @@ from .cohere_api_retriever import CohereAPIRetriever from .colidefics3_retriever import ColIdefics3Retriever from .colpali_retriever import ColPaliRetriever -from .colqwen2_retriever import ColQwen2Retriever from .colqwen2_5_retriever import ColQwen2_5_Retriever +from .colqwen2_retriever import ColQwen2Retriever from .dse_qwen2_retriever import DSEQwen2Retriever from .dummy_vision_retriever import DummyVisionRetriever from .hf_api_retriever import HFEndpointRetriever diff --git a/src/vidore_benchmark/retrievers/colqwen2_5_retriever.py b/src/vidore_benchmark/retrievers/colqwen2_5_retriever.py index 26bd1648..5be55421 100644 --- a/src/vidore_benchmark/retrievers/colqwen2_5_retriever.py +++ b/src/vidore_benchmark/retrievers/colqwen2_5_retriever.py @@ -21,7 +21,7 @@ @register_vision_retriever("colqwen2_5") -class ColQwen2_5_Retriever(BaseVisionRetriever): +class ColQwen2_5_Retriever(BaseVisionRetriever): #noqa: N801 """ ColQwen2.5 retriever that implements the model from "ColPali: Efficient Document Retrieval with Vision Language Models".