From 052e38547378d3dbf04c7c4406596d382fe9154c Mon Sep 17 00:00:00 2001
From: nguentrungthat <entjtj2000@gmail.com>
Date: Tue, 25 Mar 2025 08:51:51 +0700
Subject: [PATCH] Update train.py

---
 src/server/train.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/server/train.py b/src/server/train.py
index 87c4146..43689f5 100755
--- a/src/server/train.py
+++ b/src/server/train.py
@@ -61,6 +61,8 @@ elif os.path.exists(image_folder_yesterday) and os.path.exists(label_folder_yest
 else:
     # print(image_folder)
     # print(label_folder)
+    with open(LOG_FILE, "a") as log:
+        log.write(f"\nDate: {today_str}. No valid image & label folder found in the last two days!")
     raise Exception("  No valid image & label folder found in the last two days!")
 
 # 🏗️ Create dataset folders