Skip to content

Commit db55ff6

Browse files
committed
Fix pre-commit
Signed-off-by: herotai214 <herotai214@gmail.com>
1 parent 2665ff6 commit db55ff6

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/v1/core/test_scheduler.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2304,8 +2304,8 @@ def _assert_right_encoder_cache_allocated(
23042304
assert not missed_hashes, (
23052305
f"Miss hashes: {missed_hashes} "
23062306
f"Existing encoder cache: {encoder_cache_manager.cached}"
2307-
)
2308-
2307+
)
2308+
23092309
for req in requests if requests is not None else []:
23102310
if req.mm_features:
23112311
mm_hashes = [f.identifier for f in req.mm_features]
@@ -3120,7 +3120,7 @@ def test_ec_connector_allocate_encoder_tokens_with_external_load(use_kv_connecto
31203120
disable_hybrid_kv_cache_manager=use_kv_connector,
31213121
)
31223122

3123-
# Limit the
3123+
# Limit the number of availiable slots of EncoderCacheManager
31243124
scheduler.encoder_cache_manager = EncoderCacheManager(cache_size=32)
31253125

31263126
# Create MM request1
@@ -3179,7 +3179,7 @@ def test_ec_connector_allocate_encoder_tokens_with_external_load(use_kv_connecto
31793179

31803180
# Encoder cache should contain mm item from request1
31813181
_assert_right_encoder_cache_allocated(
3182-
scheduler, hashes_to_check=['hash1_1', 'hash2_1', 'hash2_2']
3182+
scheduler, hashes_to_check=["hash1_1", "hash2_1", "hash2_2"]
31833183
)
31843184

31853185
# request2's 2nd mm item is the last call of update_state_after_alloc

0 commit comments

Comments
 (0)