def spark_session(self, application_name): sc = self.spark_context(application_name) from pyspark.sql import SparkSession return SparkSession(sc)