fix error on machines with pyspark installed where passed dataframe i…

…s not spark pandas (#1415)
microsoft · May 11, 2022 · d0047f1 · d0047f1
1 parent 0dfa112
commit d0047f1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/erroranalysis/erroranalysis/_internal/utils.py b/erroranalysis/erroranalysis/_internal/utils.py
@@ -6,7 +6,7 @@
 import numpy as np
 
 try:
-    import pyspark
+    import pyspark.pandas as ps
     spark_available = True
 except ImportError:
     spark_available = False
@@ -47,4 +47,7 @@ def is_spark(df):
     :return: True if the dataframe is a spark dataframe, False otherwise.
     :rtype: bool
     """
-    return spark_available and isinstance(df, pyspark.pandas.frame.DataFrame)
+    try:
+        return spark_available and isinstance(df, ps.frame.DataFrame)
+    except Exception:
+        return False
diff --git a/erroranalysis/erroranalysis/version.py b/erroranalysis/erroranalysis/version.py
@@ -4,5 +4,5 @@
 name = 'erroranalysis'
 _major = '0'
 _minor = '3'
-_patch = '0'
+_patch = '1'
 version = '{}.{}.{}'.format(_major, _minor, _patch)