Skip to content

Commit

Permalink
Update holiday_features.py
Browse files Browse the repository at this point in the history
Signed-off-by: Frank Kreuwel <[email protected]>
  • Loading branch information
FrankKr authored Jan 8, 2024
1 parent 69e288e commit 0d8dfad
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion openstef/feature_engineering/holiday_features.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,14 @@ def generate_holiday_feature_functions(
"""
if years is None:
now = datetime.now()
years = [now.year - 4, now.year - 3, now.year - 2, now.year - 1, now.year]
years = [
now.year - 4,
now.year - 3,
now.year - 2,
now.year - 1,
now.year,
now.year + 1
]

country_holidays = holidays.country_holidays(country, years=years)

Expand Down

0 comments on commit 0d8dfad

Please sign in to comment.