From eea31b8cbad107401228cd932768b62e0726846b Mon Sep 17 00:00:00 2001 From: Jee Jee Li Date: Sun, 5 Jan 2025 00:07:59 +0800 Subject: [PATCH] [Bugfix][V1] Fix test_kv_cache_utils.py (#11738) Signed-off-by: Jee Jee Li --- tests/v1/core/test_kv_cache_utils.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/v1/core/test_kv_cache_utils.py b/tests/v1/core/test_kv_cache_utils.py index faa3a91de151f..2ed70b42991b5 100644 --- a/tests/v1/core/test_kv_cache_utils.py +++ b/tests/v1/core/test_kv_cache_utils.py @@ -147,12 +147,12 @@ def test_generate_block_hash_extra_keys(): # Test with no extra keys extra_keys, next_mm_idx = generate_block_hash_extra_keys(request, 0, 5, 0) - assert extra_keys == (("hash1", 0), ) + assert extra_keys == ("hash1", ) assert next_mm_idx == 1 # Test with partial overlap extra_keys, next_mm_idx = generate_block_hash_extra_keys(request, 3, 8, 0) - assert extra_keys == (("hash1", 3), ) + assert extra_keys == ("hash1", ) assert next_mm_idx == 1 # Test with no overlap @@ -162,7 +162,7 @@ def test_generate_block_hash_extra_keys(): # Test with multiple extra keys extra_keys, next_mm_idx = generate_block_hash_extra_keys(request, 0, 15, 0) - assert extra_keys == (("hash1", 0), ("hash2", 0)) + assert extra_keys == ('hash1', 'hash2') assert next_mm_idx == 2 @@ -216,11 +216,11 @@ def test_hash_request_tokens(): # Check the first block assert block_hashes[0].token_ids == (0, 1, 2) - assert block_hashes[0].extra_keys == (("hash1", 0), ) + assert block_hashes[0].extra_keys == ("hash1", ) # Check the second block assert block_hashes[1].token_ids == (3, 4, 5) - assert block_hashes[1].extra_keys == (("hash2", 0), ) + assert block_hashes[1].extra_keys == ("hash2", ) def test_hash_request_tokens_no_mm_inputs():