# It sometimes returns two dates if it is exactly half a year after the # holiday. In this case, the smaller distance (182 days) is returned. return (index - holiday_date[0]).days return distance_to_day EasterSunday = Holiday("Easter Sunday", month=1, day=1, offset=[Easter(), Day(0)]) NewYearsDay = Holiday("New Years Day", month=1, day=1) SuperBowl = Holiday("Superbowl", month=2, day=1, offset=DateOffset(weekday=SU(1))) MothersDay = Holiday("Mothers Day", month=5, day=1, offset=DateOffset(weekday=SU(2))) IndependenceDay = Holiday("Independence Day", month=7, day=4) ChristmasEve = Holiday("Christmas", month=12, day=24) ChristmasDay = Holiday("Christmas", month=12, day=25) NewYearsEve = Holiday("New Years Eve", month=12, day=31) BlackFriday = Holiday("Black Friday", month=11, day=1, offset=[pd.DateOffset(weekday=TH(4)), Day(1)]) CyberMonday = Holiday( "Cyber Monday",
index - pd.Timedelta(days=MAX_WINDOW), index + pd.Timedelta(days=MAX_WINDOW), ) assert ( len(holiday_date) != 0 ), f"No closest holiday for the date index {index} found." # It sometimes returns two dates if it is exactly half a year after the # holiday. In this case, the smaller distance (182 days) is returned. return (index - holiday_date[0]).days return distance_to_day EasterSunday = Holiday("Easter Sunday", month=1, day=1, offset=[Easter(), Day(0)]) NewYearsDay = Holiday("New Years Day", month=1, day=1) SuperBowl = Holiday("Superbowl", month=2, day=1, offset=DateOffset(weekday=SU(1))) MothersDay = Holiday("Mothers Day", month=5, day=1, offset=DateOffset(weekday=SU(2))) IndependenceDay = Holiday("Independence Day", month=7, day=4) ChristmasEve = Holiday("Christmas", month=12, day=24) ChristmasDay = Holiday("Christmas", month=12, day=25) NewYearsEve = Holiday("New Years Eve", month=12, day=31) BlackFriday = Holiday( "Black Friday", month=11, day=1, offset=[pd.DateOffset(weekday=TH(4)), Day(1)] ) CyberMonday = Holiday( "Cyber Monday", month=11, day=1, offset=[pd.DateOffset(weekday=TH(4)), Day(4)], )
assert ( len(holiday_date) != 0 ), f"No closest holiday for the date index {index} found." # It sometimes returns two dates if it is exactly half a year after the # holiday. In this case, the smaller distance (182 days) is returned. return (index - holiday_date[0]).days return distance_to_day EasterSunday = Holiday( "Easter Sunday", month=1, day=1, offset=[Easter(), Day(0)] ) NewYearsDay = Holiday("New Years Day", month=1, day=1) SuperBowl = Holiday( "Superbowl", month=2, day=1, offset=DateOffset(weekday=SU(1)) ) MothersDay = Holiday( "Mothers Day", month=5, day=1, offset=DateOffset(weekday=SU(2)) ) IndependenceDay = Holiday("Independence Day", month=7, day=4) ChristmasEve = Holiday("Christmas", month=12, day=24) ChristmasDay = Holiday("Christmas", month=12, day=25) NewYearsEve = Holiday("New Years Eve", month=12, day=31) BlackFriday = Holiday( "Black Friday", month=11, day=1, offset=[pd.DateOffset(weekday=TH(4)), Day(1)], ) CyberMonday = Holiday(