From 73df2e2a18389a8a8f892c7a368becc0c91c7d9f Mon Sep 17 00:00:00 2001
From: Faisal Dosani <faisal.dosani@pm.me>
Date: Thu, 20 Jun 2024 12:30:20 -0400
Subject: [PATCH] adding import

---
 datacompy/spark/sql.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/datacompy/spark/sql.py b/datacompy/spark/sql.py
index 961ea9d8..921b9410 100644
--- 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