From 3d42b834c970c0fc160a57b7b8b5ba6971528110 Mon Sep 17 00:00:00 2001 From: Cedar Date: Wed, 27 Nov 2024 17:36:38 -0800 Subject: [PATCH] remove not-very-useful test case --- .../llm/components/kvcache/trie_attention_cache_test.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/shortfin/tests/apps/llm/components/kvcache/trie_attention_cache_test.py b/shortfin/tests/apps/llm/components/kvcache/trie_attention_cache_test.py index 6d216c790..ce0025419 100644 --- a/shortfin/tests/apps/llm/components/kvcache/trie_attention_cache_test.py +++ b/shortfin/tests/apps/llm/components/kvcache/trie_attention_cache_test.py @@ -428,10 +428,3 @@ def test_reference_counting(trie_cache, ref_count): print("\nCleaning up allocations...") for alloc in allocations + fill_allocations: alloc.release_pages() - - -@pytest.mark.parametrize("tokens_per_page", [0, -1, -100]) -def test_invalid_init(page_pool, tokens_per_page): - """Test validation in __init__""" - with pytest.raises(ValueError): - TriePagedAttentionCache(page_pool=page_pool, tokens_per_page=tokens_per_page)