aboutsummaryrefslogtreecommitdiff
path: root/python/pyspark/ml/common.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/pyspark/ml/common.py')
-rw-r--r--python/pyspark/ml/common.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/python/pyspark/ml/common.py b/python/pyspark/ml/common.py
index 7d449aaccb..aec860fca7 100644
--- a/python/pyspark/ml/common.py
+++ b/python/pyspark/ml/common.py
@@ -51,6 +51,7 @@ py4j.protocol.smart_decode = _new_smart_decode
_picklable_classes = [
'SparseVector',
'DenseVector',
+ 'SparseMatrix',
'DenseMatrix',
]