From 1a076b2d5b5ea260775fe3a5fa86adb5125b1345 Mon Sep 17 00:00:00 2001 From: Shing Chan Date: Sat, 9 Nov 2024 16:43:54 +0000 Subject: [PATCH 1/2] decorate merge functions in merge.py with set_module('pandas') --- pandas/core/reshape/merge.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/pandas/core/reshape/merge.py b/pandas/core/reshape/merge.py index 0ca8661ad3b5c..6f9bb8cb24f43 100644 --- a/pandas/core/reshape/merge.py +++ b/pandas/core/reshape/merge.py @@ -39,7 +39,10 @@ npt, ) from pandas.errors import MergeError -from pandas.util._decorators import cache_readonly +from pandas.util._decorators import ( + cache_readonly, + set_module, +) from pandas.util._exceptions import find_stack_level from pandas.core.dtypes.base import ExtensionDtype @@ -138,6 +141,7 @@ _known = (np.ndarray, ExtensionArray, Index, ABCSeries) +@set_module("pandas") def merge( left: DataFrame | Series, right: DataFrame | Series, @@ -502,6 +506,7 @@ def _groupby_and_merge( return result, lby +@set_module("pandas") def merge_ordered( left: DataFrame | Series, right: DataFrame | Series, @@ -645,6 +650,7 @@ def _merger(x, y) -> DataFrame: return result +@set_module("pandas") def merge_asof( left: DataFrame | Series, right: DataFrame | Series, From 2c69c954b22a64ca6d4b5429dec59c5221b5f634 Mon Sep 17 00:00:00 2001 From: Shing Chan Date: Sat, 9 Nov 2024 17:04:09 +0000 Subject: [PATCH 2/2] add asserts to test merge function module string is set to 'pandas' --- pandas/tests/api/test_api.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pandas/tests/api/test_api.py b/pandas/tests/api/test_api.py index 842fa1a151267..5af2ac99030c7 100644 --- a/pandas/tests/api/test_api.py +++ b/pandas/tests/api/test_api.py @@ -416,3 +416,6 @@ def test_set_module(): assert pd.Period.__module__ == "pandas" assert pd.Timestamp.__module__ == "pandas" assert pd.Timedelta.__module__ == "pandas" + assert pd.merge.__module__ == "pandas" + assert pd.merge_ordered.__module__ == "pandas" + assert pd.merge_asof.__module__ == "pandas"