Prepare for pandas 2.0 by casting datetime to timestamp.

PiperOrigin-RevId: 576154444
tensorflow · Oct 24, 2023 · 346cb6f · 346cb6f
1 parent abdc777
commit 346cb6f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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')