Skip to content

Commit d742094

Browse files
BUG: [ArrowStringArray] Recognize ArrowStringArray in infer_dtype (#40725)
1 parent 4ccb0ff commit d742094

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

pandas/_libs/lib.pyx

+1
Original file line numberDiff line numberDiff line change
@@ -1111,6 +1111,7 @@ _TYPE_MAP = {
11111111
"complex128": "complex",
11121112
"c": "complex",
11131113
"string": "string",
1114+
str: "string",
11141115
"S": "bytes",
11151116
"U": "string",
11161117
"bool": "boolean",

pandas/tests/dtypes/test_inference.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1267,9 +1267,9 @@ def test_interval(self):
12671267
@pytest.mark.parametrize("klass", [pd.array, Series])
12681268
@pytest.mark.parametrize("skipna", [True, False])
12691269
@pytest.mark.parametrize("data", [["a", "b", "c"], ["a", "b", pd.NA]])
1270-
def test_string_dtype(self, data, skipna, klass):
1270+
def test_string_dtype(self, data, skipna, klass, nullable_string_dtype):
12711271
# StringArray
1272-
val = klass(data, dtype="string")
1272+
val = klass(data, dtype=nullable_string_dtype)
12731273
inferred = lib.infer_dtype(val, skipna=skipna)
12741274
assert inferred == "string"
12751275

0 commit comments

Comments
 (0)