address comments

apache · Apr 17, 2018 · 9a4a0ca · 9a4a0ca
1 parent b91dbeb
commit 9a4a0ca
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/pyspark/ml/stat.py b/python/pyspark/ml/stat.py
@@ -365,8 +365,8 @@ class SummaryBuilder(JavaWrapper):
     .. versionadded:: 2.4.0
 
     """
-    def __init__(self, js):
-        self._js = js
+    def __init__(self, jSummaryBuilder):
+        super(SummaryBuilder, self).__init__(jSummaryBuilder)
 
     @since("2.4.0")
     def summary(self, featuresCol, weightCol=None):
@@ -383,7 +383,7 @@ def summary(self, featuresCol, weightCol=None):
          structure is determined during the creation of the builder.
         """
         featuresCol, weightCol = Summarizer._check_param(featuresCol, weightCol)
-        return Column(self._js.summary(featuresCol._jc, weightCol._jc))
+        return Column(self._java_obj.summary(featuresCol._jc, weightCol._jc))
 
 
 if __name__ == "__main__":