Skip to content

GitLab

Unverified Commit f088027e authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[Marigold tests] add `is_flaky` decorator to some Marigold tests (#8696) 

okay
parent c6e08ecd
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
tests/pipelines/marigold/test_marigold_depth.py
View file @ f088027e
...@@ -34,6 +34,7 @@ from diffusers import ( ...@@ -34,6 +34,7 @@ from diffusers import (
from diffusers.utils.testing_utils import ( from diffusers.utils.testing_utils import (
enable_full_determinism, enable_full_determinism,
floats_tensor, floats_tensor,
is_flaky,
load_image, load_image,
require_torch_gpu, require_torch_gpu,
slow, slow,
...@@ -232,6 +233,7 @@ class MarigoldDepthPipelineFastTests(PipelineTesterMixin, unittest.TestCase): ...@@ -232,6 +233,7 @@ class MarigoldDepthPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
match_input_resolution=True, match_input_resolution=True,
) )
@is_flaky
def test_marigold_depth_dummy_G0_S1_P32_E3_B1_M1(self): def test_marigold_depth_dummy_G0_S1_P32_E3_B1_M1(self):
self._test_marigold_depth( self._test_marigold_depth(
generator_seed=0, generator_seed=0,
...@@ -244,6 +246,7 @@ class MarigoldDepthPipelineFastTests(PipelineTesterMixin, unittest.TestCase): ...@@ -244,6 +246,7 @@ class MarigoldDepthPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
match_input_resolution=True, match_input_resolution=True,
) )
@is_flaky
def test_marigold_depth_dummy_G0_S1_P32_E4_B2_M1(self): def test_marigold_depth_dummy_G0_S1_P32_E4_B2_M1(self):
self._test_marigold_depth( self._test_marigold_depth(
generator_seed=0, generator_seed=0,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment