From 08f84d7a9a7429b3d6651b5af4d7740027b53d39 Mon Sep 17 00:00:00 2001
From: Yanbo Liang <ybliang8@gmail.com>
Date: Wed, 20 Apr 2016 20:06:27 +0200
Subject: [PATCH] [MINOR][ML][PYSPARK] Fix omissive param setters which should
use _set method
## What changes were proposed in this pull request?
#11939 make Python param setters use the `_set` method. This PR fix omissive ones.
## How was this patch tested?
Existing tests.
cc jkbradley sethah
Author: Yanbo Liang <ybliang8@gmail.com>
Closes #12531 from yanboliang/setters-omissive.
---
python/pyspark/ml/feature.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/python/pyspark/ml/feature.py b/python/pyspark/ml/feature.py
index a1911cebe3..1b298e639d 100644
--- a/python/pyspark/ml/feature.py
+++ b/python/pyspark/ml/feature.py
@@ -340,7 +340,7 @@ class CountVectorizer(JavaEstimator, HasInputCol, HasOutputCol, JavaMLReadable,
"""
Sets the value of :py:attr:`binary`.
"""
- self._paramMap[self.binary] = value
+ self._set(binary=value)
return self
@since("2.0.0")
@@ -569,7 +569,7 @@ class HashingTF(JavaTransformer, HasInputCol, HasOutputCol, HasNumFeatures, Java
"""
Sets the value of :py:attr:`binary`.
"""
- self._paramMap[self.binary] = value
+ self._set(binary=value)
return self
@since("2.0.0")
--
GitLab