From 88749480d078f4e208fa61a9cd05c73a8205fbc3 Mon Sep 17 00:00:00 2001
From: Stephane Chavin <stephane.chavin@lis-lab.fr>
Date: Sun, 21 Jul 2024 19:20:42 +0200
Subject: [PATCH] Update file get_train_val.py

---
 get_train_val.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/get_train_val.py b/get_train_val.py
index 5955912..0666da9 100755
--- a/get_train_val.py
+++ b/get_train_val.py
@@ -100,7 +100,7 @@ if __name__ == '__main__':
     train, val = utils.split(df, 'train', args.ratio)
     
     saved_directory = os.path.join(args.directory, 'set')
-
+    utils.create_directory(saved_directory)
     if args.test:
         val, test = utils.split(val, 'val', 0.5)
         export_split(args, [train, val, test], args.path_to_data,
-- 
GitLab