Skip to content

Commit 48907c3

Browse files
fix for build without pyarrow
1 parent 7ffb08f commit 48907c3

File tree

1 file changed

+14
-8
lines changed

1 file changed

+14
-8
lines changed

pandas/tests/arrays/string_/test_string.py

+14-8
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010

1111
from pandas._config import using_string_dtype
1212

13+
from pandas.compat import HAS_PYARROW
1314
from pandas.compat.pyarrow import (
1415
pa_version_under12p0,
1516
pa_version_under19p0,
@@ -45,15 +46,20 @@ def cls(dtype):
4546
return dtype.construct_array_type()
4647

4748

48-
DTYPE_HIERARCHY = [
49-
pd.StringDtype("python", na_value=np.nan),
50-
pd.StringDtype("pyarrow", na_value=np.nan),
51-
pd.StringDtype("python", na_value=pd.NA),
52-
pd.StringDtype("pyarrow", na_value=pd.NA),
53-
]
54-
55-
5649
def string_dtype_highest_priority(dtype1, dtype2):
50+
if HAS_PYARROW:
51+
DTYPE_HIERARCHY = [
52+
pd.StringDtype("python", na_value=np.nan),
53+
pd.StringDtype("pyarrow", na_value=np.nan),
54+
pd.StringDtype("python", na_value=pd.NA),
55+
pd.StringDtype("pyarrow", na_value=pd.NA),
56+
]
57+
else:
58+
DTYPE_HIERARCHY = [
59+
pd.StringDtype("python", na_value=np.nan),
60+
pd.StringDtype("python", na_value=pd.NA),
61+
]
62+
5763
h1 = DTYPE_HIERARCHY.index(dtype1)
5864
h2 = DTYPE_HIERARCHY.index(dtype2)
5965
return DTYPE_HIERARCHY[max(h1, h2)]

0 commit comments

Comments
 (0)