diff options
-rw-r--r-- | python/pyspark/ml/common.py | 1 |
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', ] |