From 7ec678eb094bb61b72d6848feaac58605d33adbc Mon Sep 17 00:00:00 2001 From: Alison Shao <54658187+alisonshao@users.noreply.github.com> Date: Tue, 16 Dec 2025 17:55:15 -0800 Subject: [PATCH] [Test] Update LoRA eviction policy tests to match current behavior (#15283) --- test/nightly/test_lora_eviction_policy.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/test/nightly/test_lora_eviction_policy.py b/test/nightly/test_lora_eviction_policy.py index 943bf411c..073df2542 100644 --- a/test/nightly/test_lora_eviction_policy.py +++ b/test/nightly/test_lora_eviction_policy.py @@ -90,13 +90,13 @@ class TestLoRAEvictionPolicy(unittest.TestCase): expected_victim="lora2", ) - def test_lru_base_model_priority(self): - """Test LRU prioritizes base model for eviction.""" + def test_lru_base_model_evicted_last(self): + """Test LRU evicts LoRA adapters before base model (None).""" self._test_eviction_policy( "lru", access_sequence=["lora1", "lora2", "lora3"], candidates={None, "lora1", "lora2", "lora3"}, - expected_victim=None, + expected_victim="lora1", ) def test_fifo_basic(self): @@ -135,13 +135,13 @@ class TestLoRAEvictionPolicy(unittest.TestCase): expected_victim="lora2", ) - def test_fifo_base_model_priority(self): - """Test FIFO prioritizes base model for eviction.""" + def test_fifo_base_model_evicted_last(self): + """Test FIFO evicts LoRA adapters before base model (None).""" self._test_eviction_policy( "fifo", access_sequence=["lora1", "lora2", "lora3"], candidates={None, "lora1", "lora2", "lora3"}, - expected_victim=None, + expected_victim="lora1", ) def test_policy_remove(self):