Skip to content

Commit f088027

Browse files
authored
[Marigold tests] add is_flaky decorator to some Marigold tests (#8696)
okay
1 parent c6e08ec commit f088027

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/pipelines/marigold/test_marigold_depth.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434
from diffusers.utils.testing_utils import (
3535
enable_full_determinism,
3636
floats_tensor,
37+
is_flaky,
3738
load_image,
3839
require_torch_gpu,
3940
slow,
@@ -232,6 +233,7 @@ def test_marigold_depth_dummy_G0_S1_P64_E1_B1_M1(self):
232233
match_input_resolution=True,
233234
)
234235

236+
@is_flaky
235237
def test_marigold_depth_dummy_G0_S1_P32_E3_B1_M1(self):
236238
self._test_marigold_depth(
237239
generator_seed=0,
@@ -244,6 +246,7 @@ def test_marigold_depth_dummy_G0_S1_P32_E3_B1_M1(self):
244246
match_input_resolution=True,
245247
)
246248

249+
@is_flaky
247250
def test_marigold_depth_dummy_G0_S1_P32_E4_B2_M1(self):
248251
self._test_marigold_depth(
249252
generator_seed=0,

0 commit comments

Comments
 (0)