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