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

Skip the sigma-gather test when diffusers is not installed

Browse source 
CI runs the backend suite without diffusers; the test checks our index math against
the scheduler's own gather, so it skips rather than fails there.
This commit is contained in:
Daniel Han 2026-07-04 05:05:54 +00:00
commit f1007fb466
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
studio/backend/tests/test_diffusion_train_perf.py
View file

@ -130,6 +130,9 @@ def test_zimage_collate_list():
# ── index-based sigma gather ──────────────────────────────────────────────────
def test_gather_sigmas_matches_search_based_gather():
# CI installs the backend test deps without diffusers; the scheduler math is what we
# are checking, so skip rather than fail there.
pytest.importorskip("diffusers")
from diffusers import FlowMatchEulerDiscreteScheduler
torch.manual_seed(0)