Address comments

apache · Jun 8, 2018 · bfcd3aa · bfcd3aa
1 parent 49323a6
commit bfcd3aa
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/pyspark/ml/__init__.py b/python/pyspark/ml/__init__.py
@@ -22,10 +22,10 @@
 from pyspark.ml.base import Estimator, Model, Transformer, UnaryTransformer
 from pyspark.ml.pipeline import Pipeline, PipelineModel
 from pyspark.ml import classification, clustering, evaluation, feature, fpm, \
-    image, pipeline, recommendation, regression, stat, tuning, util
+    image, pipeline, recommendation, regression, stat, tuning, util, linalg, param
 
 __all__ = [
     "Transformer", "UnaryTransformer", "Estimator", "Model", "Pipeline", "PipelineModel",
-    "base", "classification", "clustering", "evaluation", "feature", "fpm",
-    "image", "pipeline", "recommendation", "regression", "stat", "tuning", "util",
+    "classification", "clustering", "evaluation", "feature", "fpm", "image",
+    "recommendation", "regression", "stat", "tuning", "util", "linalg", "param",
 ]