From e0fb4770286ee954d6a58584db9503f07d6a6c30 Mon Sep 17 00:00:00 2001 From: tazlin Date: Thu, 7 Mar 2024 09:58:31 -0500 Subject: [PATCH] tests: corrects non existing lora test to new logic --- tests/model_managers/test_mm_lora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/model_managers/test_mm_lora.py b/tests/model_managers/test_mm_lora.py index 40c8b549..138137f5 100644 --- a/tests/model_managers/test_mm_lora.py +++ b/tests/model_managers/test_mm_lora.py @@ -220,7 +220,7 @@ def test_adhoc_non_existing_intstring_large(self): lora_model_manager.wait_for_adhoc_reset(15) lora_name = "99999999999999" lora_key = lora_model_manager.fetch_adhoc_lora(lora_name) - assert lora_model_manager.total_retries_attempted == 0 + assert lora_model_manager.total_retries_attempted == 1 assert lora_key is None assert not lora_model_manager.is_model_available(lora_name) lora_model_manager.stop_all()