vectorize x_years_ago_at_date

Signed-off-by: Victor Garcia Reolid <[email protected]>
SeitaBV · Mar 21, 2024 · ddfd6a5 · ddfd6a5
1 parent 6aa339b
commit ddfd6a5
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/timely_beliefs/sensors/func_store/knowledge_horizons.py b/timely_beliefs/sensors/func_store/knowledge_horizons.py
@@ -69,11 +69,8 @@ def x_years_ago_at_date(
     if isinstance(event_start, datetime):
         return x_years_ago_at_date_datetime(event_start, day, month, x, z)
     else:
-        return event_start.map(
-            lambda _event_start: x_years_ago_at_date_datetime(
-                _event_start, day, month, x, z
-            )
-        )
+        ref = event_start.to_period("1Y").to_timestamp().tz_localize(event_start.tz) + pd.DateOffset(month=month, day=day)
+        return event_start - ref
 
 
 def ex_post(