adding import

capitalone · Jun 20, 2024 · 73df2e2 · 73df2e2
1 parent 0417485
commit 73df2e2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/datacompy/spark/sql.py b/datacompy/spark/sql.py
@@ -211,6 +211,8 @@ def _validate_dataframe(
         dataframe = getattr(self, index)
 
         if __version__ >= "3.4.0":
+            import pyspark.sql.connect.dataframe
+
             instances = (pyspark.sql.DataFrame, pyspark.sql.connect.dataframe.DataFrame)
         else:
             instances = pyspark.sql.DataFrame