From 8b871775e6473980562ce046cc760c9033447634 Mon Sep 17 00:00:00 2001 From: emmatveev Date: Sun, 24 Nov 2024 13:04:18 +0300 Subject: [PATCH] fix --- app/storage/mongo/configs.py | 1 + app/storage/mongo/experiments.py | 1 + 2 files changed, 2 insertions(+) diff --git a/app/storage/mongo/configs.py b/app/storage/mongo/configs.py index e3e374a..154bd59 100644 --- a/app/storage/mongo/configs.py +++ b/app/storage/mongo/configs.py @@ -35,4 +35,5 @@ async def get(project: str, stage: str) -> list[Config]: result = [] async for item in collection.find({'stage': stage, 'project': project}): result.append(Config.model_validate(item)) + result[-1]._id = item['_id'] return result diff --git a/app/storage/mongo/experiments.py b/app/storage/mongo/experiments.py index eba4246..ed3f33f 100644 --- a/app/storage/mongo/experiments.py +++ b/app/storage/mongo/experiments.py @@ -36,4 +36,5 @@ async def get(project: str, stage: str) -> list[Experiment]: result = [] async for item in collection.find({'stage': stage, 'project': project}): result.append(Experiment.model_validate(item)) + result[-1]._id = item['_id'] return result