Skip to content
Snippets Groups Projects
Commit 1e099ad2 authored by Matthew K Defenderfer's avatar Matthew K Defenderfer
Browse files

Wrap timedelta as list

parent 4f947685
No related branches found
No related tags found
2 merge requests!76Patch v0.5.1,!69Wrap timedelta as list
...@@ -19,6 +19,8 @@ def create_timedelta_breakpoints( ...@@ -19,6 +19,8 @@ def create_timedelta_breakpoints(
delta_vals: int | List[int], delta_vals: int | List[int],
delta_unit: Literal['D','W'] delta_unit: Literal['D','W']
) -> List[np.datetime64]: ) -> List[np.datetime64]:
if not isinstance(delta_vals,list):
delta_vals=[delta_vals]
run_date = as_datetime(run_date) run_date = as_datetime(run_date)
return [run_date - as_timedelta(c,delta_unit) for c in delta_vals if c > 0] return [run_date - as_timedelta(c,delta_unit) for c in delta_vals if c > 0]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment