address a comment

apache · Jun 8, 2024 · 9bd8aea · 9bd8aea
1 parent ab95e64
commit 9bd8aea
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/python/pyspark/sql/streaming/listener.py b/python/pyspark/sql/streaming/listener.py
@@ -64,19 +64,17 @@ class StreamingQueryListener(ABC):
     """
 
     def _set_spark_session(
-        self, spark: "SparkSession"  # type: ignore[name-defined] # noqa: F821
+        self, session: "SparkSession"  # type: ignore[name-defined] # noqa: F821
     ) -> None:
-        self._sparkSession = spark
+        if self.spark is not None:
+            self.spark = session
 
     @property
     def spark(self) -> Optional["SparkSession"]:  # type: ignore[name-defined] # noqa: F821
-        if hasattr(self, "_sparkSession"):
-            return self._sparkSession
-        else:
-            return None
+        return getattr(self, "_sparkSession", None)
 
     @spark.setter
-    def spark(self, session):
+    def spark(self, session) -> None:
         # For backward compatibility
         self._sparkSession = session