fixup

Author: HyukjinKwon

python/pyspark/ml/tests/connect/test_connect_function.py

@@ -19,7 +19,7 @@
 
 from pyspark.util import is_remote_only
 from pyspark.sql import SparkSession as PySparkSession
-from pyspark.sql.dataframe import DataFrame as SDF
+from pyspark.sql.classic.dataframe import DataFrame as SDF
 from pyspark.ml import functions as SF
 from pyspark.testing.sqlutils import SQLTestUtils
 from pyspark.testing.connectutils import (

python/pyspark/sql/tests/connect/test_connect_column.py

@@ -65,7 +65,7 @@
 
 class SparkConnectColumnTests(SparkConnectSQLTestCase):
     def compare_by_show(self, df1, df2, n: int = 20, truncate: int = 20):
-        from pyspark.sql.dataframe import DataFrame as SDF
+        from pyspark.sql.classic.dataframe import DataFrame as SDF
         from pyspark.sql.connect.dataframe import DataFrame as CDF
 
         assert isinstance(df1, (SDF, CDF))

python/pyspark/sql/tests/connect/test_connect_function.py

@@ -32,7 +32,7 @@
     from pyspark.sql.connect.column import Column
     from pyspark.sql import functions as SF
     from pyspark.sql.window import Window as SW
-    from pyspark.sql.dataframe import DataFrame as SDF
+    from pyspark.sql.classic.dataframe import DataFrame as SDF
     from pyspark.sql.connect import functions as CF
     from pyspark.sql.connect.window import Window as CW
     from pyspark.sql.connect.dataframe import DataFrame as CDF