From a9b777acc424e840a66ee7968482ecbfaaf0ece0 Mon Sep 17 00:00:00 2001
From: Oleksandr Bezdieniezhnykh <oleksandr.bez@2c2p.com>
Date: Sat, 8 Jun 2024 19:34:17 +0300
Subject: [PATCH] small refactoring

---
 train.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/train.py b/train.py
index b724869..27083a5 100644
--- a/train.py
+++ b/train.py
@@ -4,8 +4,9 @@ from datetime import datetime
 from pathlib import Path
 
 from ultralytics import YOLOv10
-from constants import current_images_dir, current_labels_dir, annotation_classes, today_dataset, prefix
+from constants import current_images_dir, current_labels_dir, annotation_classes, prefix
 
+today_dataset = os.path.join('datasets', f'{prefix}{datetime.now():%Y-%m-%d}')
 yaml_path = os.path.join(today_dataset, 'data.yaml')
 train_set = 70
 valid_set = 20