Skip to content
Snippets Groups Projects

Bugfix: Added numpy datetime as valid type for run_date

1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
@@ -48,7 +48,7 @@ class Aggregator:
self,
delta_vals: int | List[int],
delta_unit: Literal['D','W','M','Y'],
run_date: pd.Timestamp
run_date: pd.Timestamp | np.datetime64
) -> List[int | pd.Timestamp]:
deltas = pd.Series([as_timedelta(c,delta_unit) for c in delta_vals])
cutoffs = pd.to_datetime(run_date - deltas)
Loading