diff --git a/python/pyspark/shell.py b/python/pyspark/shell.py index ac5ce87a3f0fddbb6d7426f30e400f9b9ef6024d..c1917d2be69d8b4819fc0c64b3275ce00ef0b148 100644 --- a/python/pyspark/shell.py +++ b/python/pyspark/shell.py @@ -49,6 +49,7 @@ except TypeError: spark = SparkSession.builder.getOrCreate() sc = spark.sparkContext +sql = spark.sql atexit.register(lambda: sc.stop()) # for compatibility