diff --git a/python/pyspark/testing/connectutils.py b/python/pyspark/testing/connectutils.py index 05df6b02e6726..7f4250613cc20 100644 --- a/python/pyspark/testing/connectutils.py +++ b/python/pyspark/testing/connectutils.py @@ -97,6 +97,7 @@ def _session_sql(cls, query: str) -> "DataFrame": return DataFrame.withPlan(SQL(query), cls.connect) # type: ignore if have_pandas: + @classmethod def _with_plan(cls, plan: LogicalPlan) -> "DataFrame": return DataFrame.withPlan(plan, cls.connect) # type: ignore