command to start sagemaker training

include sample training

src/main.py

@@ -6,7 +6,7 @@ from rich.console import Console
 from rich.progress import BarColumn, Progress, SpinnerColumn, TaskProgressColumn, TextColumn
 
 from src.aws import s3 as s3_ops
-from src.commands import infra
+from src.commands import infra, train
 from src.commands.utils import CONFIG_OPT, load_cfg
 from src.config import Config
 
@@ -15,6 +15,7 @@ app = typer.Typer(
     no_args_is_help=True,
 )
 app.add_typer(infra.app, name="infra")
+app.add_typer(train.app, name="train")
 
 console = Console()
 
@@ -36,7 +37,10 @@ def init(
         yaml.safe_dump(config.model_dump(mode="json"), f, sort_keys=False)
 
     console.print(f"[green]✓[/green] Config written to [bold]{dest}[/bold]")
-    console.print("Edit it (especially [cyan]s3.bucket[/cyan]) before running other commands.")
+    console.print(
+        "Edit it (especially [cyan]s3.bucket[/cyan] and [cyan]sagemaker.training.image_uri[/cyan]) "
+        "before running other commands."
+    )
 
 
 @app.command()