Skip to content

Commit ab2817c

Browse files
committed
fix the string datatype
1 parent 084b199 commit ab2817c

File tree

3 files changed

+10
-2
lines changed

3 files changed

+10
-2
lines changed

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -137,3 +137,4 @@ doc/source/savefig/
137137
# Interactive terminal generated files #
138138
########################################
139139
.jupyterlite.doit.db
140+
.venv

pandas/core/arrays/string_.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@
6464
from pandas.core.construction import extract_array
6565
from pandas.core.indexers import check_array_indexer
6666
from pandas.core.missing import isna
67-
67+
from pandas.util._decorators import set_module
6868
from pandas.io.formats import printing
6969

7070
if TYPE_CHECKING:
@@ -86,6 +86,7 @@
8686
from pandas import Series
8787

8888

89+
@set_module("pandas")
8990
@register_extension_dtype
9091
class StringDtype(StorageExtensionDtype):
9192
"""
@@ -373,7 +374,7 @@ def __from_arrow__(
373374
NDArrayBacked.__init__(new_string_array, arr, self)
374375
return new_string_array
375376

376-
377+
@set_module("pandas")
377378
class BaseStringArray(ExtensionArray):
378379
"""
379380
Mixin class for StringArray, ArrowStringArray.
@@ -532,6 +533,7 @@ def _str_map_nan_semantics(
532533

533534
# error: Definition of "_concat_same_type" in base class "NDArrayBacked" is
534535
# incompatible with definition in base class "ExtensionArray"
536+
535537
class StringArray(BaseStringArray, NumpyExtensionArray): # type: ignore[misc]
536538
"""
537539
Extension array for string data.
@@ -959,6 +961,7 @@ def _cmp_method(self, other, op):
959961
_arith_method = _cmp_method
960962

961963

964+
962965
class StringArrayNumpySemantics(StringArray):
963966
_storage = "python"
964967
_na_value = np.nan

pandas/tests/api/test_api.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -416,3 +416,7 @@ def test_set_module():
416416
assert pd.Period.__module__ == "pandas"
417417
assert pd.Timestamp.__module__ == "pandas"
418418
assert pd.Timedelta.__module__ == "pandas"
419+
assert pd.StringDtype.__module__ == "pandas"
420+
assert pd.BaseStringArray.__module__ == "pandas"
421+
assert pd.StringArrayNumpySemantics.___module__ == "pandas"
422+
assert pd.StringArray.__module__ == "pandas"

0 commit comments

Comments
 (0)