ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-05-15 12:57:37 +00:00 • committed by danielhanchen
commit 961d83ae53
1 changed files with 4 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/studio/run_real_mlx_smoke.py
View file

@ -371,9 +371,7 @@ def cmd_train(args) -> int:
# don't gate on it -- the post_train_loss assertion above is the
# real memorisation gate, and the lora / merged / gguf reload paths
# below each have their own soft-checked generation assertion.
metrics["in_memory_generation_has_expected"] = (
EXPECT_IN_OUTPUT in in_mem_out
)
metrics["in_memory_generation_has_expected"] = EXPECT_IN_OUTPUT in in_mem_out
if EXPECT_IN_OUTPUT not in in_mem_out:
print(
f" [WARN] in-memory completion did not contain "
@ -527,9 +525,9 @@ def cmd_reload(args) -> int:
else:
# Fallback when train_metrics.json wasn't found (older
# workdir layouts): keep the original gibberish gate.
assert EXPECT_IN_OUTPUT in out, (
f"reload {args.format!r} produced gibberish for {PROMPT!r}: {out!r}"
)
assert (
EXPECT_IN_OUTPUT in out
), f"reload {args.format!r} produced gibberish for {PROMPT!r}: {out!r}"
metrics["final_peak_gpu_gb"] = round(_peak_gpu_gb(), 3)
metrics["final_peak_rss_gb"] = round(_peak_rss_gb(), 3)