From 36b2e1cfee26d097e4de5541cf27ad8887e9fb60 Mon Sep 17 00:00:00 2001
From: Maxime <672982+maximegmd@users.noreply.github.com>
Date: Tue, 29 Aug 2023 15:17:10 +0200
Subject: [PATCH] tweak: use default config file when only one file is present
 (#501)

---
 scripts/finetune.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/scripts/finetune.py b/scripts/finetune.py
index 454a627a1..044199c16 100644
--- a/scripts/finetune.py
+++ b/scripts/finetune.py
@@ -153,6 +153,10 @@ def choose_config(path: Path):
             "No YAML config files found in the specified directory. Are you using a .yml extension?"
         )
 
+    if len(yaml_files) == 1:
+        print(f"Using default YAML file '{yaml_files[0]}'")
+        return yaml_files[0]
+
     print("Choose a YAML file:")
     for idx, file in enumerate(yaml_files):
         print(f"{idx + 1}. {file}")