Skip to content

Commit e5f9c4e

Browse files
committed
[bugfix] tests: include ltx2_3_base in local LTX-2 preset set
hao-ai-lab#1428 fixed this exact assertion in fastvideo/tests/api/test_presets.py but missed the identical one in tests/local_tests/. Since local_tests are excluded from CI, test_ltx2_typed_surface_preflight has been failing since ltx2_3_base was registered in hao-ai-lab#1397.
1 parent 9212f4f commit e5f9c4e

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/local_tests/ltx2/test_ltx2_pipeline_smoke.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -316,9 +316,9 @@ def test_ltx2_typed_surface_preflight() -> None:
316316
LTX2AudioDecodingStage, LTX2DenoisingStage, LTX2LatentPreparationStage, LTX2TextEncodingStage,
317317
)
318318

319-
# All three LTX-2 presets registered.
319+
# All four LTX-2 presets registered.
320320
names = {p.name for p in get_presets_for_family("ltx2")}
321-
assert names == {"ltx2_base", "ltx2_distilled", "ltx2_two_stage"}
321+
assert names == {"ltx2_base", "ltx2_3_base", "ltx2_distilled", "ltx2_two_stage"}
322322

323323
# Two-stage preset has the denoise + refine topology and pulls its
324324
# refine allowed_overrides from the typed dataclass.

0 commit comments

Comments
 (0)