diff options
Diffstat (limited to 'python/pyspark/__init__.py')
-rw-r--r-- | python/pyspark/__init__.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/python/pyspark/__init__.py b/python/pyspark/__init__.py index 949406c57b..9f71db397d 100644 --- a/python/pyspark/__init__.py +++ b/python/pyspark/__init__.py @@ -42,7 +42,7 @@ from pyspark.context import SparkContext from pyspark.rdd import RDD from pyspark.files import SparkFiles from pyspark.storagelevel import StorageLevel -from pyspark.mllib import train_linear_regression_model +from pyspark.mllib import LinearRegressionModel -__all__ = ["SparkContext", "RDD", "SparkFiles", "StorageLevel", "train_linear_regression_model"] +__all__ = ["SparkContext", "RDD", "SparkFiles", "StorageLevel", "LinearRegressionModel"]; |