From 90f6cd417d2f60c09f7d9c26c5a2a3e990861a1a Mon Sep 17 00:00:00 2001 From: LeoBorcherding Date: Thu, 21 May 2026 02:10:27 -0500 Subject: [PATCH] fix(tests): correct MLX smoke test losses_per_step assertion logging_steps=1 with max_steps=30 produces 30 loss entries, not 7. The assertion was stale from a previous config. --- tests/studio/run_real_mlx_smoke.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/studio/run_real_mlx_smoke.py b/tests/studio/run_real_mlx_smoke.py index 27f682ee4e..9a18313152 100644 --- a/tests/studio/run_real_mlx_smoke.py +++ b/tests/studio/run_real_mlx_smoke.py @@ -390,7 +390,7 @@ def cmd_train(args) -> int: ) if k in train_result } - assert len(losses_per_step) == 7, f"expected 7 logged steps, got {losses_per_step}" + assert len(losses_per_step) == 30, f"expected 30 logged steps (logging_steps=1, max_steps=30), got {losses_per_step}" for i, l in enumerate(losses_per_step): # Allow exact 0.0: fp16 per-step loss underflows to 0.0 after # the LoRA reaches loss=0 around step ~10 with this fixture +