From 7abf62f877b075956fbd7dbbbb3f60e452ca9907 Mon Sep 17 00:00:00 2001 From: Marco Gaido Date: Fri, 26 Jan 2018 17:07:45 +0100 Subject: [PATCH] [SPARK-23234][ML][PYSPARK] Remove setting defaults on java params --- python/pyspark/ml/wrapper.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/python/pyspark/ml/wrapper.py b/python/pyspark/ml/wrapper.py index 0f846fbc5b5e..6d263d4daa26 100644 --- a/python/pyspark/ml/wrapper.py +++ b/python/pyspark/ml/wrapper.py @@ -118,10 +118,9 @@ def _transfer_params_to_java(self): """ Transforms the embedded params to the companion Java object. """ - paramMap = self.extractParamMap() for param in self.params: - if param in paramMap: - pair = self._make_java_param_pair(param, paramMap[param]) + if param in self._paramMap: + pair = self._make_java_param_pair(param, self._paramMap[param]) self._java_obj.set(pair) def _transfer_param_map_to_java(self, pyParamMap):