Added array datatypes to shorthand list

markgw · Aug 5, 2019 · 9a4d98f · 9a4d98f
1 parent 258bd05
commit 9a4d98f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git a/src/python/pimlico/datatypes/__init__.py b/src/python/pimlico/datatypes/__init__.py
@@ -11,13 +11,14 @@
 from .gensim import GensimLdaModel
 from .corpora.base import IterableCorpus
 from .corpora.grouped import GroupedCorpus
+from .arrays import NumpyArray, ScipySparseMatrix
 
 # All builtin datatypes that may be easily loaded using their
 # class names of datatype names from config files
 BUILTIN_DATATYPES = [
     PimlicoDatatype, IterableCorpus, GroupedCorpus,
     StringList, Dict, NamedFileCollection, NamedFile, TextFile, Dictionary,
-    Embeddings, TSVVecFiles, GensimLdaModel,
+    Embeddings, TSVVecFiles, GensimLdaModel, NumpyArray, ScipySparseMatrix,
 ]
 BUILTIN_DATATYPES_BY_DATATYPE_NAME = dict(
     # Go through them in reverse, so that, if we make a mistake and have a duplicate name, the

diff --git a/src/python/pimlico/datatypes/arrays.py b/src/python/pimlico/datatypes/arrays.py
@@ -5,12 +5,6 @@
 """
 Wrappers around Numpy arrays and Scipy sparse matrices.
 
-.. todo::
-
-   During Python 2-3 conversion, an ``object`` base class was added to ``NumpyArray.Reader``
-   and ``Writer`` and ``ScipySparseMatrix.Reader`` and ``Writer``.
-   Check that these still work as they used to.
-
 """
 
 from builtins import object