From 03f5a7fd16222aef66d3dffa80ef63a5f77f06e0 Mon Sep 17 00:00:00 2001
From: Salman Mohammadi <salman.mohammadi@outlook.com>
Date: Tue, 12 Aug 2025 18:35:12 +0100
Subject: [PATCH] adding back

---
 src/axolotl/cli/utils/train.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/axolotl/cli/utils/train.py b/src/axolotl/cli/utils/train.py
index ddcf5a68f..b8c87d86c 100644
--- a/src/axolotl/cli/utils/train.py
+++ b/src/axolotl/cli/utils/train.py
@@ -72,7 +72,9 @@ def generate_config_files(config: str, sweep: str | None) -> Iterator[tuple[str,
     Args:
         config: Base configuration file
         sweep: Sweep configuration file
-
+        
+    Yields:
+        tuple[str, bool]: Configuration file name and whether this is a group of configurations
     """
 
     if not sweep: