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

Pin the accelerator probes in the DiT family-metadata tests

Browse source 
Six tests read family_train_infos() (or a start preflight) without pinning the
host probes, so they only held on a machine with a bf16 accelerator: on a
GPU-less runner the DiT gate empties precision_modes, turns supports_compile
off, and replaces any other preflight message with the no-accelerator note, and
all six failed there. A conftest fixture pins both probes for exactly those
tests, so they assert the family metadata they are about on every host. The
gate's own CPU-only behaviour keeps its dedicated tests.
This commit is contained in:
Daniel Han 2026-07-26 21:41:28 +00:00
commit 4e32ef6484
5 changed files with 23 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

4
studio/backend/tests/test_diffusion_base_precision.py
View file

@ -92,7 +92,7 @@ def test_base_precision_denies_fp8_for_corrupted_family():
assert flux.base_precision == "fp8"
def test_family_train_infos_drops_denied_fp8_for_qwen(monkeypatch):
def test_family_train_infos_drops_denied_fp8_for_qwen(monkeypatch, dit_train_host):
# /info advertises the machine's DiT modes per family, but a family whose DiT the mode corrupts
# must not offer it, so the UI never surfaces a mode normalized() would reject.
monkeypatch.setattr(
@ -528,7 +528,7 @@ def test_train_precision_modes_gates_dense_on_bf16_support(monkeypatch):
# ── family_train_infos precision fields ───────────────────────────────────────
def test_family_train_infos_carries_precision_fields(monkeypatch):
def test_family_train_infos_carries_precision_fields(monkeypatch, dit_train_host):
# Pin the machine probe so the DiT families carry a deterministic mode list, while SDXL (no
# precision selector) stays empty regardless.
monkeypatch.setattr(common, "train_precision_modes", lambda: (["nf4", "bf16"], "auto"))