diff --git a/src/axolotl/cli/utils/train.py b/src/axolotl/cli/utils/train.py
index c266dab2e..5ab6b5156 100644
--- a/src/axolotl/cli/utils/train.py
+++ b/src/axolotl/cli/utils/train.py
@@ -74,9 +74,7 @@ def generate_config_files(config: str, sweep: str | None) -> Iterator[tuple[str,
         sweep: Sweep configuration file
 
     Yields:
-        tuple[str, bool]: (config_file, is_group)
-            - config_file: configuration file path
-            - is_group: whether this is a group of configurations
+        tuple[str, bool]: configuration file path, and a boolean indicating if this is a group of configurations.
     """
 
     if not sweep: