File tree Expand file tree Collapse file tree 1 file changed +7
-2
lines changed Expand file tree Collapse file tree 1 file changed +7
-2
lines changed Original file line number Diff line number Diff line change 11# SPDX-License-Identifier: Apache-2.0
22# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3+ import dataclasses
34from typing import Optional
45from unittest .mock import Mock
56
@@ -1921,8 +1922,12 @@ def test_chunked_prefill_disabled_for_encoder_decoder(
19211922 is_encoder_decoder = is_encoder_decoder ,
19221923 )
19231924
1924- # `is_encoder_decoder` should only be used during construction of the config
1925- assert not hasattr (scheduler_config , "is_encoder_decoder" )
1925+ # `is_encoder_decoder` should only be used during construction
1926+ # of the config, and otherwise stored in the model config.
1927+ assert "is_encoder_decoder" not in vars (scheduler_config )
1928+ assert "is_encoder_decoder" not in [
1929+ f .name for f in dataclasses .fields (scheduler_config )
1930+ ]
19261931 _validate_chunked_prefill_settings_for_encoder_decoder (
19271932 scheduler_config , is_encoder_decoder , expect_enabled )
19281933
You can’t perform that action at this time.
0 commit comments