[TEST] Fix wrong image model in image gen samplers test

Aedial · May 25, 2023 · 39aeab9 · 39aeab9
1 parent 44db083
commit 39aeab9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/api/test_imagegen_samplers.py b/tests/api/test_imagegen_samplers.py
@@ -14,7 +14,9 @@
 sampler_xfail = pytest.mark.xfail(True, raises=NovelAIError, reason="The sampler doesn't currently work")
 
 models = list(ImageModel)
-models.remove(ImageModel.Anime_Inpainting)
+models.remove(ImageModel.Inpainting_Anime_Full)
+models.remove(ImageModel.Inainting_Anime_Curated)
+models.remove(ImageModel.Inpainting_Furry)
 
 samplers = list(ImageSampler)
 model_samplers = list(itertools.product(models, samplers))