Skip to content

Commit d3ad7b0

Browse files
remove hardcoded storage
1 parent 864c166 commit d3ad7b0

File tree

4 files changed

+8
-7
lines changed

4 files changed

+8
-7
lines changed

pandas/core/dtypes/cast.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -801,7 +801,7 @@ def infer_dtype_from_scalar(val) -> tuple[DtypeObj, Any]:
801801
if using_string_dtype():
802802
from pandas.core.arrays.string_ import StringDtype
803803

804-
dtype = StringDtype(storage="pyarrow", na_value=np.nan)
804+
dtype = StringDtype(na_value=np.nan)
805805

806806
elif isinstance(val, (np.datetime64, dt.datetime)):
807807
try:

pandas/core/internals/construction.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -302,7 +302,7 @@ def ndarray_to_mgr(
302302
nb = new_block_2d(values, placement=bp, refs=refs)
303303
block_values = [nb]
304304
elif dtype is None and values.dtype.kind == "U" and using_string_dtype():
305-
dtype = StringDtype(storage="pyarrow", na_value=np.nan)
305+
dtype = StringDtype(na_value=np.nan)
306306

307307
obj_columns = list(values)
308308
block_values = [

pandas/io/_util.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,6 @@ def arrow_string_types_mapper() -> Callable:
3434
pa = import_optional_dependency("pyarrow")
3535

3636
return {
37-
pa.string(): pd.StringDtype(storage="pyarrow", na_value=np.nan),
38-
pa.large_string(): pd.StringDtype(storage="pyarrow", na_value=np.nan),
37+
pa.string(): pd.StringDtype(na_value=np.nan),
38+
pa.large_string(): pd.StringDtype(na_value=np.nan),
3939
}.get

pandas/io/pytables.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,7 @@
7575
PeriodIndex,
7676
RangeIndex,
7777
Series,
78+
StringDtype,
7879
TimedeltaIndex,
7980
concat,
8081
isna,
@@ -3295,7 +3296,7 @@ def read(
32953296
values = self.read_array("values", start=start, stop=stop)
32963297
result = Series(values, index=index, name=self.name, copy=False)
32973298
if using_string_dtype() and is_string_array(values, skipna=True):
3298-
result = result.astype("string[pyarrow_numpy]")
3299+
result = result.astype(StringDtype(na_value=np.nan))
32993300
return result
33003301

33013302
def write(self, obj, **kwargs) -> None:
@@ -3364,7 +3365,7 @@ def read(
33643365
columns = items[items.get_indexer(blk_items)]
33653366
df = DataFrame(values.T, columns=columns, index=axes[1], copy=False)
33663367
if using_string_dtype() and is_string_array(values, skipna=True):
3367-
df = df.astype("string[pyarrow_numpy]")
3368+
df = df.astype(StringDtype(na_value=np.nan))
33683369
dfs.append(df)
33693370

33703371
if len(dfs) > 0:
@@ -4741,7 +4742,7 @@ def read(
47414742
values, # type: ignore[arg-type]
47424743
skipna=True,
47434744
):
4744-
df = df.astype("string[pyarrow_numpy]")
4745+
df = df.astype(StringDtype(na_value=np.nan))
47454746
frames.append(df)
47464747

47474748
if len(frames) == 1:

0 commit comments

Comments
 (0)