From 35002ad21ecb408ced2a96eb09f3a85fc02475ce Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Xueguang=20Ma=20=E9=A9=AC=E9=9B=AA=E5=85=89?= Date: Mon, 4 Dec 2023 09:36:41 -0500 Subject: [PATCH] add missing import CosDprDocumentEncoder (#1738) * add missing import --- pyserini/encode/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyserini/encode/__init__.py b/pyserini/encode/__init__.py index 20abf9c06..bcbcafa00 100644 --- a/pyserini/encode/__init__.py +++ b/pyserini/encode/__init__.py @@ -27,4 +27,4 @@ from ._splade import SpladeQueryEncoder from ._slim import SlimQueryEncoder from ._openai import OpenAIDocumentEncoder, OpenAIQueryEncoder, OPENAI_API_RETRY_DELAY -from ._cosdpr import CosDprEncoder, CosDprQueryEncoder \ No newline at end of file +from ._cosdpr import CosDprEncoder, CosDprDocumentEncoder, CosDprQueryEncoder