fix test_names LoraManager test

finegrain-ai · Jan 23, 2024 · 3b458f0 · 3b458f0
1 parent 421da6a
commit 3b458f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: