diff --git a/studio/backend/tests/test_diffusion_transformer_quant.py b/studio/backend/tests/test_diffusion_transformer_quant.py index 03cea6c53a..aac5251084 100644 --- a/studio/backend/tests/test_diffusion_transformer_quant.py +++ b/studio/backend/tests/test_diffusion_transformer_quant.py @@ -365,7 +365,6 @@ def test_exclude_tokens_for_scheme_shared_by_runtime_and_builder(): _INT8_EXCLUDE_NAME_TOKENS, exclude_tokens_for_scheme, ) - assert exclude_tokens_for_scheme(TQ_INT8) == _INT8_EXCLUDE_NAME_TOKENS for scheme in (TQ_FP8, TQ_NVFP4, TQ_MXFP8): assert exclude_tokens_for_scheme(scheme) == ()