review comments

Author: grundprinzip

python/pyspark/sql/connect/dataframe.py

@@ -2208,15 +2208,6 @@ def rdd(self) -> "RDD[Row]":
 
     @property
     def queryExecution(self) -> Optional["QueryExecution"]:
-        """
-        The queryExecution method allows to introspect information about the actual
-        query execution after the successful execution. Accessing this member before
-        the query execution has happened will return None.
-
-        Returns
-        -------
-        An instance of QueryExecution or None when the value is not set yet.
-        """
         return self._query_execution
 
 

python/pyspark/sql/dataframe.py

@@ -6284,6 +6284,19 @@ def toPandas(self) -> "PandasDataFrameLike":
 
     @property
     def queryExecution(self) -> Optional["QueryExecution"]:
+        """
+        Returns a QueryExecution object after the query was executed.
+
+        The queryExecution method allows to introspect information about the actual
+        query execution after the successful execution. Accessing this member before
+        the query execution will return None.
+
+        .. versionadded:: 4.0.0
+
+        Returns
+        -------
+        An instance of QueryExecution or None when the value is not set yet.
+        """
         ...
 
 

python/pyspark/sql/metrics.py

@@ -19,9 +19,10 @@
 from typing import Optional, List, Tuple, Dict, Any, Union, TYPE_CHECKING, Sequence
 
 from pyspark.errors import PySparkValueError
-from pyspark.testing.connectutils import have_graphviz
 
 if TYPE_CHECKING:
+    from pyspark.testing.connectutils import have_graphviz
+
     if have_graphviz:
         import graphviz  # type: ignore