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: