From deaee145520691cd9b14fc9294f09b47e2343afb Mon Sep 17 00:00:00 2001 From: ssmall41 Date: Sat, 30 May 2020 10:39:39 +0200 Subject: [PATCH] Add imports from holiday.py in unit test module (#14) Fixes failing unit tests due to missing imports. --- test/feature/test_holiday.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/test/feature/test_holiday.py b/test/feature/test_holiday.py index 582901f..6c582bb 100644 --- a/test/feature/test_holiday.py +++ b/test/feature/test_holiday.py @@ -41,7 +41,9 @@ from pts.feature.holiday import ( CYBER_MONDAY, SpecialDateFeatureSet, squared_exponential_kernel, - CustomDateFeatureSet + exponential_kernel, + CustomDateFeatureSet, + CustomHolidayFeatureSet ) test_dates = {