diff --git a/aligned/local/job.py b/aligned/local/job.py index 3a199cb..e550508 100644 --- a/aligned/local/job.py +++ b/aligned/local/job.py @@ -83,6 +83,7 @@ async def aggregate(request: RetrivalRequest, core_data: pl.LazyFrame) -> pl.Laz time_name, every=over.window.every_interval, period=over.window.time_window, + offset=over.window.offset_interval, by=over.group_by_names, ) .agg(exprs) @@ -90,7 +91,10 @@ async def aggregate(request: RetrivalRequest, core_data: pl.LazyFrame) -> pl.Laz ) else: sub = sorted_data.groupby_rolling( - time_name, period=over.window.time_window, by=over.group_by_names + time_name, + period=over.window.time_window, + offset=over.window.offset_interval, + by=over.group_by_names, ).agg(exprs) if results is not None: diff --git a/pyproject.toml b/pyproject.toml index 2428b8b..180ee49 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "aligned" -version = "0.0.44" +version = "0.0.45" description = "A scalable feature store that makes it easy to align offline and online ML systems" authors = ["Mats E. Mollestad "] license = "Apache-2.0"