From c2e1428360a1c47ca2c1478aeb806aff986e4e1c Mon Sep 17 00:00:00 2001 From: Sara H Date: Wed, 30 Oct 2024 15:00:06 -0500 Subject: [PATCH] extract variable from with --- backend/app/infrastructure/repositories/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/app/infrastructure/repositories/model.py b/backend/app/infrastructure/repositories/model.py index 7013bb5b..177dc720 100644 --- a/backend/app/infrastructure/repositories/model.py +++ b/backend/app/infrastructure/repositories/model.py @@ -40,9 +40,9 @@ def get_model_in_the_loop(self, task_id: int) -> dict: return models_in_the_loop def update_light_model(self, id: int, light_model: str) -> None: + light_model = f"{light_model}/model/single_evaluation" with self.session as session: instance = session.query(self.model).filter(self.model.id == id).first() - light_model = f"{light_model}/model/single_evaluation" instance.light_model = light_model instance.is_in_the_loop = 0 session.flush()