From 37bed97de5f81a9127f1ff12db695aab6e5c4c47 Mon Sep 17 00:00:00 2001
From: WeichenXu <WeichenXu123@outlook.com>
Date: Sun, 24 Jul 2016 02:29:08 -0700
Subject: [PATCH] [PYSPARK] add picklable SparseMatrix in pyspark.ml.common
## What changes were proposed in this pull request?
add `SparseMatrix` class whick support pickler.
## How was this patch tested?
Existing test.
Author: WeichenXu <WeichenXu123@outlook.com>
Closes #14265 from WeichenXu123/picklable_py.
---
python/pyspark/ml/common.py | 1 +
1 file changed, 1 insertion(+)
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',
]
--
GitLab