Added Matrix, SparseMatrix to __all__ list in linalg.py

jeanlyn · Apr 28, 2015 · deb51a2 · deb51a2
1 parent 28b1af7
commit deb51a2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/pyspark/mllib/linalg.py b/python/pyspark/mllib/linalg.py
@@ -39,7 +39,8 @@
     IntegerType, ByteType
 
 
-__all__ = ['Vector', 'DenseVector', 'SparseVector', 'Vectors', 'DenseMatrix', 'Matrices']
+__all__ = ['Vector', 'DenseVector', 'SparseVector', 'Vectors',
+           'Matrix', 'DenseMatrix', 'SparseMatrix', 'Matrices']
 
 
 if sys.version_info[:2] == (2, 7):