Fix LTX2 channels-last review scope

Signed-off-by: Hongsheng Liu <liuhongsheng4@huawei.com>

tests/diffusion/models/ltx2/test_ltx2_vae.py

@@ -12,6 +12,21 @@ pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
 
 
 class TestLTXChannelsLast3D:
+    @pytest.mark.parametrize(
+        ("extras", "expected"),
+        [({}, False), ({"ltx2_use_channels_last_3d": True}, True)],
+    )
+    def test_channels_last_opt_in_is_ltx2_model_extra(self, extras, expected):
+        from vllm_omni.diffusion.models.ltx2.ltx2_components import _ltx2_use_channels_last_3d
+
+        assert _ltx2_use_channels_last_3d(SimpleNamespace(extras=extras)) is expected
+
+    def test_channels_last_opt_in_rejects_non_boolean_model_extra(self):
+        from vllm_omni.diffusion.models.ltx2.ltx2_components import _ltx2_use_channels_last_3d
+
+        with pytest.raises(TypeError, match="ltx2_use_channels_last_3d"):
+            _ltx2_use_channels_last_3d(SimpleNamespace(extras={"ltx2_use_channels_last_3d": "true"}))
+
     @pytest.mark.parametrize(
         ("causal", "kernel_size"),
         [

tests/diffusion/test_diffusion_engine.py

@@ -534,18 +534,6 @@ class TestDiffusionCompileConfig:
         with pytest.raises(TypeError, match="diffusion_compile_dynamic"):
             OmniDiffusionConfig(model="test", diffusion_compile_dynamic="false")
 
-    def test_channels_last_3d_is_opt_in_and_strict(self) -> None:
-        assert OmniDiffusionConfig(model="test").vae_use_channels_last_3d is False
-        assert (
-            OmniDiffusionConfig.from_kwargs(
-                model="test",
-                vae_use_channels_last_3d=True,
-            ).vae_use_channels_last_3d
-            is True
-        )
-        with pytest.raises(TypeError, match="vae_use_channels_last_3d"):
-            OmniDiffusionConfig(model="test", vae_use_channels_last_3d="true")
-
     @pytest.mark.parametrize(
         "kwargs, feature",
         [