Fix detecting connect availability

G-Research · Aug 15, 2024 · 2362923 · 2362923
1 parent 4cbd9b0
commit 2362923
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/python/gresearch/spark/__init__.py b/python/gresearch/spark/__init__.py
@@ -39,7 +39,7 @@
     from pyspark.sql.connect.readwriter import DataFrameReader as ConnectDataFrameReader
     from pyspark.sql.connect.session import SparkSession as ConnectSparkSession
     has_connect = True
-except ModuleNotFoundError:
+except ImportError:
     has_connect = False
 
 if TYPE_CHECKING:

diff --git a/python/gresearch/spark/diff/__init__.py b/python/gresearch/spark/diff/__init__.py
@@ -26,7 +26,7 @@
 try:
     from pyspark.sql.connect.dataframe import DataFrame as ConnectDataFrame
     has_connect = True
-except ModuleNotFoundError:
+except ImportError:
     has_connect = False
 
 

diff --git a/python/gresearch/spark/parquet/__init__.py b/python/gresearch/spark/parquet/__init__.py
@@ -22,7 +22,7 @@
 try:
     from pyspark.sql.connect.readwriter import DataFrameReader as ConnectDataFrameReader
     has_connect = True
-except ModuleNotFoundError:
+except ImportError:
     has_connect = False