tests: skip sde check during CI

Haidra-Org · Feb 19, 2024 · 8d47c20 · 8d47c20
1 parent 2c8ebe2
commit 8d47c20
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/tests/test_horde_samplers.py b/tests/test_horde_samplers.py
@@ -122,10 +122,15 @@ def test_samplers(
 
         for img_filename, pil_image in images_to_compare:
             logger.debug(f"Checking image {img_filename}")
-            assert check_single_inference_image_similarity(
-                img_filename,
-                pil_image,
-            )
+            if "sde" not in img_filename:
+                assert check_single_inference_image_similarity(
+                    img_filename,
+                    pil_image,
+                )
+            else:
+                logger.warning(
+                    f"Skipping image similarity check for {img_filename} due to SDE samplers being non-deterministic.",
+                )
 
     def test_slow_samplers(
         self,