aboutsummaryrefslogtreecommitdiff
path: root/examples/src/main/python/ml/onehot_encoder_example.py
diff options
context:
space:
mode:
Diffstat (limited to 'examples/src/main/python/ml/onehot_encoder_example.py')
-rw-r--r--examples/src/main/python/ml/onehot_encoder_example.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/examples/src/main/python/ml/onehot_encoder_example.py b/examples/src/main/python/ml/onehot_encoder_example.py
index 9acc363dc9..b9fceef68e 100644
--- a/examples/src/main/python/ml/onehot_encoder_example.py
+++ b/examples/src/main/python/ml/onehot_encoder_example.py
@@ -23,7 +23,10 @@ from pyspark.ml.feature import OneHotEncoder, StringIndexer
from pyspark.sql import SparkSession
if __name__ == "__main__":
- spark = SparkSession.builder.appName("OneHotEncoderExample").getOrCreate()
+ spark = SparkSession\
+ .builder\
+ .appName("OneHotEncoderExample")\
+ .getOrCreate()
# $example on$
df = spark.createDataFrame([