aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--python/pyspark/ml/pipeline.py2
-rw-r--r--python/pyspark/ml/wrapper.py3
2 files changed, 3 insertions, 2 deletions
diff --git a/python/pyspark/ml/pipeline.py b/python/pyspark/ml/pipeline.py
index 4475451edb..9f5f6ac8fa 100644
--- a/python/pyspark/ml/pipeline.py
+++ b/python/pyspark/ml/pipeline.py
@@ -86,7 +86,7 @@ class Transformer(Params):
@abstractmethod
def _transform(self, dataset):
"""
- Transforms the input dataset with optional parameters.
+ Transforms the input dataset.
:param dataset: input dataset, which is an instance of
:py:class:`pyspark.sql.DataFrame`
diff --git a/python/pyspark/ml/wrapper.py b/python/pyspark/ml/wrapper.py
index 4bcb4aaec8..dd1d4b076e 100644
--- a/python/pyspark/ml/wrapper.py
+++ b/python/pyspark/ml/wrapper.py
@@ -15,7 +15,7 @@
# limitations under the License.
#
-from abc import ABCMeta
+from abc import ABCMeta, abstractmethod
from pyspark import SparkContext
from pyspark.sql import DataFrame
@@ -110,6 +110,7 @@ class JavaEstimator(Estimator, JavaWrapper):
__metaclass__ = ABCMeta
+ @abstractmethod
def _create_model(self, java_model):
"""
Creates a model from the input Java model reference.