explicit calls to param checks

apache · Feb 8, 2018 · 339c793 · 339c793
1 parent bbd2144
commit 339c793
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/pyspark/ml/wrapper.py b/python/pyspark/ml/wrapper.py
@@ -119,10 +119,10 @@ def _transfer_params_to_java(self):
         Transforms the embedded params to the companion Java object.
         """
         for param in self.params:
-            if param in self._paramMap:
+            if self.isSet(param):
                 pair = self._make_java_param_pair(param, self._paramMap[param])
                 self._java_obj.set(pair)
-            if param in self._defaultParamMap:
+            if self.hasDefault(param):
                 pair = self._make_java_param_pair(param, self._defaultParamMap[param])
                 self._java_obj.setDefaultPair(pair)