Skip to content

Commit 2abc703

Browse files
Make test compatible with pydantic.dataclasses
Signed-off-by: simondanielsson <simon.danielsson99@hotmail.com>
1 parent 9b1a2a8 commit 2abc703

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

tests/v1/core/test_scheduler.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
# SPDX-License-Identifier: Apache-2.0
22
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3+
import dataclasses
34
from typing import Optional
45
from 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

0 commit comments

Comments
 (0)