CLI fix for backend changes

This commit is contained in:
Manan17 2026-03-09 07:00:06 +00:00
commit b8aa528cf3
2 changed files with 11 additions and 6 deletions

View file

@ -22,8 +22,11 @@ def list_checkpoints(
typer.echo("No checkpoints found.")
raise typer.Exit()
for display, path in checkpoints:
typer.echo(f"{display}: {path}")
for model_name, ckpt_list, metadata in checkpoints:
typer.echo(f"\n{model_name}:")
for display, path, loss in ckpt_list:
loss_str = f" (loss: {loss:.4f})" if loss is not None else ""
typer.echo(f" {display}{loss_str}: {path}")
def export(