diff --git a/tensorflow_probability/python/sts/holiday_effects.py b/tensorflow_probability/python/sts/holiday_effects.py index e6c23aaa79..571fadbf7c 100644 --- a/tensorflow_probability/python/sts/holiday_effects.py +++ b/tensorflow_probability/python/sts/holiday_effects.py @@ -52,8 +52,8 @@ def get_default_holidays(times, country): columns=['geo', 'holiday', 'date']) holidays = holidays.explode('holiday') # Ensure that only holiday dates covered by times are used. - holidays = holidays[(holidays['date'] >= times.min()) - & (holidays['date'] <= times.max())] + holidays = holidays[(pd.to_datetime(holidays['date']) >= times.min()) + & (pd.to_datetime(holidays['date']) <= times.max())] holidays = holidays.reset_index(drop=True) holidays['date'] = pd.to_datetime(holidays['date']) holidays = holidays.sort_values('date')