diff options
-rw-r--r-- | python/pyspark/ml/pipeline.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/python/pyspark/ml/pipeline.py b/python/pyspark/ml/pipeline.py index 4307ad02a0..a78e3b49fb 100644 --- a/python/pyspark/ml/pipeline.py +++ b/python/pyspark/ml/pipeline.py @@ -21,7 +21,7 @@ if sys.version > '3': basestring = str from pyspark import since, keyword_only, SparkContext -from pyspark.ml import Estimator, Model, Transformer +from pyspark.ml.base import Estimator, Model, Transformer from pyspark.ml.param import Param, Params from pyspark.ml.util import JavaMLWriter, JavaMLReader, MLReadable, MLWritable from pyspark.ml.wrapper import JavaParams |