tasks - save chip length when preprocessing steps is finished

hotosm · May 30, 2024 · e784173 · e784173
1 parent d10c647
commit e784173
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/backend/core/tasks.py b/backend/core/tasks.py
@@ -206,6 +206,8 @@ def train_model(
             training_instance.chips_length = get_file_count(
                 os.path.join(preprocess_output, "chips")
             )
+            training_instance.save()
+
             # train
 
             train_output = f"{base_path}/train"