Skip to content
Snippets Groups Projects
Commit bf491bb3 authored by Tor Myklebust's avatar Tor Myklebust
Browse files

The rest of the Python side of those bindings.

parent 95915f8b
No related branches found
No related tags found
No related merge requests found
......@@ -42,6 +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
__all__ = ["SparkContext", "RDD", "SparkFiles", "StorageLevel"]
__all__ = ["SparkContext", "RDD", "SparkFiles", "StorageLevel", "train_linear_regression_model"]
......@@ -62,5 +62,6 @@ def launch_gateway():
# Import the classes used by PySpark
java_import(gateway.jvm, "org.apache.spark.api.java.*")
java_import(gateway.jvm, "org.apache.spark.api.python.*")
java_import(gateway.jvm, "org.apache.spark.mllib.api.*")
java_import(gateway.jvm, "scala.Tuple2")
return gateway
......@@ -308,4 +308,4 @@ def write_int(value, stream):
def write_with_length(obj, stream):
write_int(len(obj), stream)
stream.write(obj)
\ No newline at end of file
stream.write(obj)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment