translate holidays to english is possible

ourownstory · Jun 22, 2024 · 60260bd · 60260bd
1 parent a436231
commit 60260bd
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/neuralprophet/event_utils.py b/neuralprophet/event_utils.py
@@ -63,7 +63,7 @@ def get_all_holidays(years, country):
         single_country = "TUR" if single_country == "TU" else single_country
         # get dict of dates and their holiday name
         single_country_specific_holidays = country_holidays(
-            country=single_country, subdiv=subdivision, years=years, expand=True, observed=False
+            country=single_country, subdiv=subdivision, years=years, expand=True, observed=False, language="en"
         )
         # invert order - for given holiday, store list of dates
         for date, name in single_country_specific_holidays.items():

diff --git a/poetry.lock b/poetry.lock
diff --git a/tests/test_unit.py b/tests/test_unit.py
@@ -1008,6 +1008,7 @@ def test_multiple_countries():
     holiday_names = m.model.config_holidays.holiday_names
     assert "Independence Day" in holiday_names
     assert "Christmas Day" in holiday_names
+    assert "Erster Weihnachtstag" not in holiday_names
     assert "Neujahr" not in holiday_names