From ad9ec515f4745e0973d6a571b91e1005cecfaf9b Mon Sep 17 00:00:00 2001 From: Aman Sharma <210100011@iitb.ac.in> Date: Sat, 16 Nov 2024 14:17:20 +0530 Subject: [PATCH 1/3] decorate concat in concat.py with set_module('pandas') --- pandas/core/reshape/concat.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pandas/core/reshape/concat.py b/pandas/core/reshape/concat.py index cfe83111b6e38..0bd8eecf64d9e 100644 --- a/pandas/core/reshape/concat.py +++ b/pandas/core/reshape/concat.py @@ -18,6 +18,7 @@ from pandas._libs import lib from pandas.util._exceptions import find_stack_level +from pandas.util._decorators import set_module from pandas.core.dtypes.common import ( is_bool, @@ -149,6 +150,7 @@ def concat( ) -> DataFrame | Series: ... +@set_module("pandas") def concat( objs: Iterable[Series | DataFrame] | Mapping[HashableT, Series | DataFrame], *, From 5a86a13c74d51a000b6d18d27a9b2b11526c037e Mon Sep 17 00:00:00 2001 From: Aman Sharma <210100011@iitb.ac.in> Date: Sat, 16 Nov 2024 14:23:18 +0530 Subject: [PATCH 2/3] add asserts to test concat function module string is set to 'pandas' --- pandas/tests/api/test_api.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pandas/tests/api/test_api.py b/pandas/tests/api/test_api.py index 75f9958b16286..c1d9f5ea4d25c 100644 --- a/pandas/tests/api/test_api.py +++ b/pandas/tests/api/test_api.py @@ -417,6 +417,7 @@ def test_set_module(): assert pd.Period.__module__ == "pandas" assert pd.Timestamp.__module__ == "pandas" assert pd.Timedelta.__module__ == "pandas" + assert pd.concat.__module__ == "pandas" assert pd.isna.__module__ == "pandas" assert pd.notna.__module__ == "pandas" assert pd.merge.__module__ == "pandas" From c8bdc46c1aa3006a598c8368b89d1eef68d584a1 Mon Sep 17 00:00:00 2001 From: Aman Sharma <210100011@iitb.ac.in> Date: Sat, 16 Nov 2024 14:42:27 +0530 Subject: [PATCH 3/3] fix whitespace error --- pandas/core/reshape/concat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pandas/core/reshape/concat.py b/pandas/core/reshape/concat.py index 0bd8eecf64d9e..e7cb7069bbc26 100644 --- a/pandas/core/reshape/concat.py +++ b/pandas/core/reshape/concat.py @@ -17,8 +17,8 @@ import numpy as np from pandas._libs import lib -from pandas.util._exceptions import find_stack_level from pandas.util._decorators import set_module +from pandas.util._exceptions import find_stack_level from pandas.core.dtypes.common import ( is_bool,