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