Skip to content

Commit

Permalink
ENH: Set __module__ on StringDtype (#60261)
Browse files Browse the repository at this point in the history
  • Loading branch information
espoirMur authored Nov 11, 2024
1 parent b434476 commit d3c595e
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 1 deletion.
6 changes: 5 additions & 1 deletion pandas/core/arrays/string_.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,10 @@
pa_version_under10p1,
)
from pandas.compat.numpy import function as nv
from pandas.util._decorators import doc
from pandas.util._decorators import (
doc,
set_module,
)
from pandas.util._exceptions import find_stack_level

from pandas.core.dtypes.base import (
Expand Down Expand Up @@ -86,6 +89,7 @@
from pandas import Series


@set_module("pandas")
@register_extension_dtype
class StringDtype(StorageExtensionDtype):
"""
Expand Down
1 change: 1 addition & 0 deletions pandas/tests/api/test_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -405,6 +405,7 @@ def test_set_module():
assert pd.IntervalDtype.__module__ == "pandas"
assert pd.SparseDtype.__module__ == "pandas"
assert pd.ArrowDtype.__module__ == "pandas"
assert pd.StringDtype.__module__ == "pandas"
assert pd.Index.__module__ == "pandas"
assert pd.CategoricalIndex.__module__ == "pandas"
assert pd.DatetimeIndex.__module__ == "pandas"
Expand Down

0 comments on commit d3c595e

Please sign in to comment.