From dbfb6ed9fd8295af16b5f48a562d2d8bbf0cc894 Mon Sep 17 00:00:00 2001 From: itqop Date: Tue, 27 Feb 2024 23:24:45 +0300 Subject: [PATCH] =?UTF-8?q?=E2=9C=85=20Fix=20pipeline:=20changed=20bounds?= =?UTF-8?q?=20in=20densify=20(all=20tests=20passed)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/database/methods.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/app/database/methods.py b/app/database/methods.py index fc1a556..1805e35 100644 --- a/app/database/methods.py +++ b/app/database/methods.py @@ -20,7 +20,7 @@ async def aggregate_salaries(collection: AsyncIOMotorCollection, dt_from: str, d "step": 1, "unit": group_type, "bounds": [ - iso(dt_from), iso(dt_upto) + iso(dt_from), iso(dt_upto) + dt.timedelta(days=int(group_type=="day"), hours=int(group_type=="hour")) * (iso(dt_upto).time() == dt.time(hour=0, minute=0, second=0)) ] } } @@ -56,4 +56,6 @@ async def aggregate_salaries(collection: AsyncIOMotorCollection, dt_from: str, d return {"dataset": dataset, "labels": labels} def iso(date: str) -> dt.datetime: - return dt.datetime.fromisoformat(date) \ No newline at end of file + return dt.datetime.fromisoformat(date) + +