Fix __reduce__ of TDigest to work with pickle/Spark broadacast (#25)

isarn · Jan 28, 2021 · 279af3b · 279af3b
1 parent 045da6a
commit 279af3b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/isarnproject/sketches/spark/tdigest.py b/python/isarnproject/sketches/spark/tdigest.py
@@ -260,7 +260,7 @@ def isEmpty(self):
         return len(self._cent) == 0
 
     def __reduce__(self):
-        return (self.__class__, (self.compression, self.maxDiscrete, self.nclusters, self._cent, self._mass, ))
+        return (self.__class__, (self.compression, self.maxDiscrete, self._cent, self._mass, ))
 
     def _lmcovj(self, m):
         assert self.nclusters >= 2