From f1007fb4662636678fdc10516812cfe08f95b3e6 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sat, 4 Jul 2026 05:05:54 +0000 Subject: [PATCH] Skip the sigma-gather test when diffusers is not installed 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. --- studio/backend/tests/test_diffusion_train_perf.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/studio/backend/tests/test_diffusion_train_perf.py b/studio/backend/tests/test_diffusion_train_perf.py index 4339fbb0e7..713f7e4dd7 100644 --- a/studio/backend/tests/test_diffusion_train_perf.py +++ b/studio/backend/tests/test_diffusion_train_perf.py @@ -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)