From deb51a2cac955b56596c92ab912475c4e3ab35b5 Mon Sep 17 00:00:00 2001
From: "Joseph K. Bradley" <joseph@databricks.com>
Date: Tue, 28 Apr 2015 14:11:41 -0700
Subject: [PATCH] Added Matrix, SparseMatrix to __all__ list in linalg.py

---
 python/pyspark/mllib/linalg.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/pyspark/mllib/linalg.py b/python/pyspark/mllib/linalg.py
index cc9a4cf8ba170..a57c0b3ae0d00 100644
--- 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):