From 3b458f0d8d7f0ee3fbe23ae47008ee19852edbc4 Mon Sep 17 00:00:00 2001 From: limiteinductive Date: Tue, 23 Jan 2024 12:52:15 +0100 Subject: [PATCH] fix test_names LoraManager test --- tests/foundationals/latent_diffusion/test_lora_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/foundationals/latent_diffusion/test_lora_manager.py b/tests/foundationals/latent_diffusion/test_lora_manager.py index e9dab1c..a126934 100644 --- a/tests/foundationals/latent_diffusion/test_lora_manager.py +++ b/tests/foundationals/latent_diffusion/test_lora_manager.py @@ -75,7 +75,7 @@ def test_names(manager: SDLoraManager, weights: dict[str, torch.Tensor]) -> None assert manager.names == ["pokemon-lora"] manager.add_loras("pokemon-lora2", tensors=weights) - assert manager.names == ["pokemon-lora", "pokemon-lora2"] + assert set(manager.names) == set(["pokemon-lora", "pokemon-lora2"]) def test_scales(manager: SDLoraManager, weights: dict[str, torch.Tensor]) -> None: