From 6ea6c95229cd2226479feafdb5c0eced6a6e2c9f Mon Sep 17 00:00:00 2001 From: Shing Chan Date: Sat, 9 Nov 2024 16:49:55 +0000 Subject: [PATCH 1/2] decorate timedelta_range with set_module('pandas') --- pandas/core/indexes/timedeltas.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pandas/core/indexes/timedeltas.py b/pandas/core/indexes/timedeltas.py index 6bbe86816d81f..fa3de46621643 100644 --- a/pandas/core/indexes/timedeltas.py +++ b/pandas/core/indexes/timedeltas.py @@ -237,6 +237,7 @@ def inferred_type(self) -> str: return "timedelta64" +@set_module("pandas") def timedelta_range( start=None, end=None, From dfaef75dea8390c1c36fabc6dcf59f07041bc60c Mon Sep 17 00:00:00 2001 From: Shing Chan Date: Sat, 9 Nov 2024 16:50:31 +0000 Subject: [PATCH 2/2] add assert to test timedelta_range module string is '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 842fa1a151267..80c2e05ce44f4 100644 --- a/pandas/tests/api/test_api.py +++ b/pandas/tests/api/test_api.py @@ -416,3 +416,4 @@ def test_set_module(): assert pd.Period.__module__ == "pandas" assert pd.Timestamp.__module__ == "pandas" assert pd.Timedelta.__module__ == "pandas" + assert pd.timedelta_range.__module__ == "pandas"