test

llmfoundry/cli/cli.py

@@ -11,10 +11,12 @@
 app.add_typer(registry_cli.app, name='registry')
 
 @app.command(name="train")
-def train(yaml_path: str = typer.Argument(..., help="Path to the YAML configuration file")):
+def train(
+    yaml_path: str = typer.Argument(..., help="Path to the YAML configuration file"),
+    additional_args: list = typer.Argument(None, help="Additional command line arguments")
+):
     """Run the training with the given configuration and optional overrides from command line."""
-    args_list = sys.argv[2:]  # assuming additional arguments may follow the yaml_path
-    run_trainer(yaml_path, args_list)
+    run_trainer(yaml_path, additional_args)
 
 if __name__ == '__main__':
     app()

llmfoundry/train/train.py

@@ -568,8 +568,6 @@ def run_trainer(
     yaml_path: str, 
     args_list: str
 ):
-    print('-----------')
-    print(yaml_path, args_list)
     # Disable resolving environment variables through omegaconf.
     om.clear_resolver('oc.env')
 
@@ -582,7 +580,6 @@ def run_trainer(
         cfg,
         DictConfig,
     ), 'Configuration must be a DictConfig instance.'
-    print(cfg)
     main(cfg)