Skip to content

Commit e531732

Browse files
authored
ENH: set __module__ on merge functions in merge.py (#60266)
1 parent d770a80 commit e531732

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

pandas/core/reshape/merge.py

+7-1
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,10 @@
3939
npt,
4040
)
4141
from pandas.errors import MergeError
42-
from pandas.util._decorators import cache_readonly
42+
from pandas.util._decorators import (
43+
cache_readonly,
44+
set_module,
45+
)
4346
from pandas.util._exceptions import find_stack_level
4447

4548
from pandas.core.dtypes.base import ExtensionDtype
@@ -138,6 +141,7 @@
138141
_known = (np.ndarray, ExtensionArray, Index, ABCSeries)
139142

140143

144+
@set_module("pandas")
141145
def merge(
142146
left: DataFrame | Series,
143147
right: DataFrame | Series,
@@ -502,6 +506,7 @@ def _groupby_and_merge(
502506
return result, lby
503507

504508

509+
@set_module("pandas")
505510
def merge_ordered(
506511
left: DataFrame | Series,
507512
right: DataFrame | Series,
@@ -645,6 +650,7 @@ def _merger(x, y) -> DataFrame:
645650
return result
646651

647652

653+
@set_module("pandas")
648654
def merge_asof(
649655
left: DataFrame | Series,
650656
right: DataFrame | Series,

pandas/tests/api/test_api.py

+3
Original file line numberDiff line numberDiff line change
@@ -417,6 +417,9 @@ def test_set_module():
417417
assert pd.Period.__module__ == "pandas"
418418
assert pd.Timestamp.__module__ == "pandas"
419419
assert pd.Timedelta.__module__ == "pandas"
420+
assert pd.merge.__module__ == "pandas"
421+
assert pd.merge_ordered.__module__ == "pandas"
422+
assert pd.merge_asof.__module__ == "pandas"
420423
assert pd.read_csv.__module__ == "pandas"
421424
assert pd.read_table.__module__ == "pandas"
422425
assert pd.read_fwf.__module__ == "pandas"

0 commit comments

Comments
 (0)