[Test] Update LoRA eviction policy tests to match current behavior (#15283)
This commit is contained in:
@@ -90,13 +90,13 @@ class TestLoRAEvictionPolicy(unittest.TestCase):
|
|||||||
expected_victim="lora2",
|
expected_victim="lora2",
|
||||||
)
|
)
|
||||||
|
|
||||||
def test_lru_base_model_priority(self):
|
def test_lru_base_model_evicted_last(self):
|
||||||
"""Test LRU prioritizes base model for eviction."""
|
"""Test LRU evicts LoRA adapters before base model (None)."""
|
||||||
self._test_eviction_policy(
|
self._test_eviction_policy(
|
||||||
"lru",
|
"lru",
|
||||||
access_sequence=["lora1", "lora2", "lora3"],
|
access_sequence=["lora1", "lora2", "lora3"],
|
||||||
candidates={None, "lora1", "lora2", "lora3"},
|
candidates={None, "lora1", "lora2", "lora3"},
|
||||||
expected_victim=None,
|
expected_victim="lora1",
|
||||||
)
|
)
|
||||||
|
|
||||||
def test_fifo_basic(self):
|
def test_fifo_basic(self):
|
||||||
@@ -135,13 +135,13 @@ class TestLoRAEvictionPolicy(unittest.TestCase):
|
|||||||
expected_victim="lora2",
|
expected_victim="lora2",
|
||||||
)
|
)
|
||||||
|
|
||||||
def test_fifo_base_model_priority(self):
|
def test_fifo_base_model_evicted_last(self):
|
||||||
"""Test FIFO prioritizes base model for eviction."""
|
"""Test FIFO evicts LoRA adapters before base model (None)."""
|
||||||
self._test_eviction_policy(
|
self._test_eviction_policy(
|
||||||
"fifo",
|
"fifo",
|
||||||
access_sequence=["lora1", "lora2", "lora3"],
|
access_sequence=["lora1", "lora2", "lora3"],
|
||||||
candidates={None, "lora1", "lora2", "lora3"},
|
candidates={None, "lora1", "lora2", "lora3"},
|
||||||
expected_victim=None,
|
expected_victim="lora1",
|
||||||
)
|
)
|
||||||
|
|
||||||
def test_policy_remove(self):
|
def test_policy_remove(self):
|
||||||
|
|||||||
Reference in New Issue
Block a user