From 924a13f724aabf47bec43227e5edd0cbc918b6b7 Mon Sep 17 00:00:00 2001
From: shamoon <4887959+shamoon@users.noreply.github.com>
Date: Tue, 29 Apr 2025 11:46:18 -0700
Subject: [PATCH] Fix: always update classifier task result (#9817)

---
 src/documents/tasks.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/documents/tasks.py b/src/documents/tasks.py
index 7d71d48c9..857ace928 100644
--- a/src/documents/tasks.py
+++ b/src/documents/tasks.py
@@ -123,14 +123,15 @@ def train_classifier(*, scheduled=True):
             task.result = "Training data unchanged"
 
         task.status = states.SUCCESS
-        task.date_done = timezone.now()
-        task.save(update_fields=["status", "result", "date_done"])
 
     except Exception as e:
         logger.warning("Classifier error: " + str(e))
         task.status = states.FAILURE
         task.result = str(e)
 
+    task.date_done = timezone.now()
+    task.save(update_fields=["status", "result", "date_done"])
+
 
 @shared_task(bind=True)
 def consume_file(