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: