diff --git a/python/pyspark/ml/common.py b/python/pyspark/ml/common.py index 7d449aaccb44fde533edd45e9b60732e02751c66..aec860fca70578eb15cfb96d8d302f3ab81f825e 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', ]