diff --git a/pandas/core/reshape/melt.py b/pandas/core/reshape/melt.py index 20b4cd2185bb4..5d4b15c9a0ca3 100644 --- a/pandas/core/reshape/melt.py +++ b/pandas/core/reshape/melt.py @@ -5,6 +5,8 @@ import numpy as np +from pandas.util._decorators import set_module + from pandas.core.dtypes.common import ( is_iterator, is_list_like, @@ -39,6 +41,7 @@ def ensure_list_vars(arg_vars, variable: str, columns) -> list: return [] +@set_module("pandas") def melt( frame: DataFrame, id_vars=None, @@ -275,6 +278,7 @@ def melt( return result +@set_module("pandas") def lreshape(data: DataFrame, groups: dict, dropna: bool = True) -> DataFrame: """ Reshape wide-format data to long. Generalized inverse of DataFrame.pivot. @@ -361,6 +365,7 @@ def lreshape(data: DataFrame, groups: dict, dropna: bool = True) -> DataFrame: return data._constructor(mdata, columns=id_cols + pivot_cols) +@set_module("pandas") def wide_to_long( df: DataFrame, stubnames, i, j, sep: str = "", suffix: str = r"\d+" ) -> DataFrame: diff --git a/pandas/core/reshape/pivot.py b/pandas/core/reshape/pivot.py index 5320e6b95e406..04c584c226aed 100644 --- a/pandas/core/reshape/pivot.py +++ b/pandas/core/reshape/pivot.py @@ -10,6 +10,7 @@ import numpy as np from pandas._libs import lib +from pandas.util._decorators import set_module from pandas.core.dtypes.cast import maybe_downcast_to_dtype from pandas.core.dtypes.common import ( @@ -50,6 +51,7 @@ from pandas import DataFrame +@set_module("pandas") def pivot_table( data: DataFrame, values=None, @@ -699,6 +701,7 @@ def _convert_by(by): return by +@set_module("pandas") def pivot( data: DataFrame, *, @@ -917,6 +920,7 @@ def pivot( return result +@set_module("pandas") def crosstab( index, columns, diff --git a/pandas/core/reshape/tile.py b/pandas/core/reshape/tile.py index 44b5df6a4f1b2..b13da83084e5c 100644 --- a/pandas/core/reshape/tile.py +++ b/pandas/core/reshape/tile.py @@ -17,6 +17,7 @@ Timestamp, lib, ) +from pandas.util._decorators import set_module from pandas.core.dtypes.common import ( ensure_platform_int, @@ -51,6 +52,7 @@ ) +@set_module("pandas") def cut( x, bins, @@ -287,6 +289,7 @@ def cut( return _postprocess_for_cut(fac, bins, retbins, original) +@set_module("pandas") def qcut( x, q, diff --git a/pandas/tests/api/test_api.py b/pandas/tests/api/test_api.py index f18e999ab0c28..849a81eaf56d9 100644 --- a/pandas/tests/api/test_api.py +++ b/pandas/tests/api/test_api.py @@ -530,5 +530,13 @@ def test_set_module(): assert pd.to_numeric.__module__ == "pandas" assert pd.NamedAgg.__module__ == "pandas" assert pd.IndexSlice.__module__ == "pandas" + assert pd.lreshape.__module__ == "pandas" + assert pd.melt.__module__ == "pandas" + assert pd.wide_to_long.__module__ == "pandas" + assert pd.crosstab.__module__ == "pandas" + assert pd.pivot_table.__module__ == "pandas" + assert pd.pivot.__module__ == "pandas" + assert pd.cut.__module__ == "pandas" + assert pd.qcut.__module__ == "pandas" assert api.typing.SeriesGroupBy.__module__ == "pandas.api.typing" assert api.typing.DataFrameGroupBy.__module__ == "pandas.api.typing"