From f088027e937b2ee1acef1f6b2776b7b2fee7ffd6 Mon Sep 17 00:00:00 2001
From: Sayak Paul <spsayakpaul@gmail.com>
Date: Tue, 25 Jun 2024 21:57:28 +0530
Subject: [PATCH] [Marigold tests] add `is_flaky` decorator to some Marigold
 tests (#8696)

okay
---
 tests/pipelines/marigold/test_marigold_depth.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/pipelines/marigold/test_marigold_depth.py b/tests/pipelines/marigold/test_marigold_depth.py
index 24d1981b8f..fcb9adca7a 100644
--- a/tests/pipelines/marigold/test_marigold_depth.py
+++ b/tests/pipelines/marigold/test_marigold_depth.py
@@ -34,6 +34,7 @@ from diffusers import (
 from diffusers.utils.testing_utils import (
     enable_full_determinism,
     floats_tensor,
+    is_flaky,
     load_image,
     require_torch_gpu,
     slow,
@@ -232,6 +233,7 @@ class MarigoldDepthPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
             match_input_resolution=True,
         )
 
+    @is_flaky
     def test_marigold_depth_dummy_G0_S1_P32_E3_B1_M1(self):
         self._test_marigold_depth(
             generator_seed=0,
@@ -244,6 +246,7 @@ class MarigoldDepthPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
             match_input_resolution=True,
         )
 
+    @is_flaky
     def test_marigold_depth_dummy_G0_S1_P32_E4_B2_M1(self):
         self._test_marigold_depth(
             generator_seed=0,