Merge pull request #169 from hotosm/fix/task_id_null

Fix : Null log
hotosm · Oct 3, 2023 · 7ec1f77 · 7ec1f77
2 parents 6fda004 + cb11701
commit 7ec1f77
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/backend/core/tasks.py b/backend/core/tasks.py
@@ -57,7 +57,9 @@ def train_model(
     try:
         ## -----------IMAGE DOWNLOADER---------
         os.makedirs(settings.LOG_PATH, exist_ok=True)
-
+        if training_instance.task_id is None or training_instance.task_id.strip() == '': 
+            training_instance.task_id=train_model.request.id
+            training_instance.save()
         log_file = os.path.join(
             settings.LOG_PATH, f"run_{train_model.request.id}_log.txt"
         )