extract variable from with

mlcommons · Oct 30, 2024 · c2e1428 · c2e1428
1 parent 033b453
commit c2e1428
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()