From 2a7930e24d14b885a7f032b6c906a8895d9b3f2f Mon Sep 17 00:00:00 2001
From: kshitijrajsharma <skshitizraj@gmail.com>
Date: Thu, 30 May 2024 09:41:55 +0000
Subject: [PATCH] tasks - save chip length when preprocessing steps is finished

---
 backend/core/tasks.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/backend/core/tasks.py b/backend/core/tasks.py
index 8fbca094..bfbf7c3b 100644
--- 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"