Fixed docs

markgw · Oct 2, 2020 · 0ca7116 · 0ca7116
1 parent 7110eb5
commit 0ca7116
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/python/pimlico/datatypes/embeddings.py b/src/python/pimlico/datatypes/embeddings.py
@@ -26,7 +26,8 @@
 from pimlico.datatypes.files import NamedFileCollection
 from pimlico.utils.core import cached_property
 
-__all__ = ["Embeddings", "TSVVecFiles", "Word2VecFiles"]
+__all__ = ["Vocab", "Embeddings", "TSVVecFiles", "Word2VecFiles",
+           "DocEmbeddingsMapper", "FixedEmbeddingsDocMapper", "FastTextDocMapper"]
 
 
 class Vocab(object):