From 9d05b68e5ce6772110e957c26140bbb642a90251 Mon Sep 17 00:00:00 2001 From: Shing Chan Date: Sat, 9 Nov 2024 16:03:29 +0000 Subject: [PATCH 1/3] decorate date_range with set_module('pandas') --- pandas/core/indexes/datetimes.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pandas/core/indexes/datetimes.py b/pandas/core/indexes/datetimes.py index b3d9c3bc78a66..f043047798fab 100644 --- a/pandas/core/indexes/datetimes.py +++ b/pandas/core/indexes/datetimes.py @@ -816,6 +816,7 @@ def indexer_between_time( return mask.nonzero()[0] +@set_module("pandas") def date_range( start=None, end=None, From dd0ccc8a309669c26aed08ce19a3db3a8c70d58f Mon Sep 17 00:00:00 2001 From: Shing Chan Date: Sat, 9 Nov 2024 16:13:58 +0000 Subject: [PATCH 2/3] decorate bdate_range with set_module('pandas') --- pandas/core/indexes/datetimes.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pandas/core/indexes/datetimes.py b/pandas/core/indexes/datetimes.py index f043047798fab..9adbaadbdcdc8 100644 --- a/pandas/core/indexes/datetimes.py +++ b/pandas/core/indexes/datetimes.py @@ -1021,6 +1021,7 @@ def date_range( return DatetimeIndex._simple_new(dtarr, name=name) +@set_module("pandas") def bdate_range( start=None, end=None, From 5f92bd1ba77f15256a7170a162315b81fdaf6f3f Mon Sep 17 00:00:00 2001 From: Shing Chan Date: Sat, 9 Nov 2024 16:21:13 +0000 Subject: [PATCH 3/3] add asserts to test module string is set for date_range and bdate_range --- pandas/tests/api/test_api.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pandas/tests/api/test_api.py b/pandas/tests/api/test_api.py index 842fa1a151267..4c7f4dd163041 100644 --- a/pandas/tests/api/test_api.py +++ b/pandas/tests/api/test_api.py @@ -416,3 +416,5 @@ def test_set_module(): assert pd.Period.__module__ == "pandas" assert pd.Timestamp.__module__ == "pandas" assert pd.Timedelta.__module__ == "pandas" + assert pd.date_range.__module__ == "pandas" + assert pd.bdate_range.__module__ == "pandas"