Skip to content

Commit d9a9cf4

Browse files
DN6sayakpaul
authored andcommitted
[CI] Skip flaky download tests in PR CI (#8945)
update
1 parent edc20c3 commit d9a9cf4

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

tests/models/test_modeling_common.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -124,6 +124,7 @@ def test_cached_files_are_used_when_no_internet(self):
124124
if p1.data.ne(p2.data).sum() > 0:
125125
assert False, "Parameters not the same!"
126126

127+
@unittest.skip("Flaky behaviour on CI. Re-enable after migrating to new runners")
127128
@unittest.skipIf(torch_device == "mps", reason="Test not supported for MPS.")
128129
def test_one_request_upon_cached(self):
129130
use_safetensors = False

tests/pipelines/test_pipelines.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -146,6 +146,7 @@ def __init__(self, encoder: CustomEncoder, scheduler: DDIMScheduler):
146146

147147

148148
class DownloadTests(unittest.TestCase):
149+
@unittest.skip("Flaky behaviour on CI. Re-enable after migrating to new runners")
149150
def test_one_request_upon_cached(self):
150151
# TODO: For some reason this test fails on MPS where no HEAD call is made.
151152
if torch_device == "mps":
@@ -191,6 +192,7 @@ def test_less_downloads_passed_object(self):
191192
assert "scheduler" in os.listdir(cached_folder)
192193
assert "feature_extractor" in os.listdir(cached_folder)
193194

195+
@unittest.skip("Flaky behaviour on CI. Re-enable after migrating to new runners")
194196
def test_less_downloads_passed_object_calls(self):
195197
# TODO: For some reason this test fails on MPS where no HEAD call is made.
196198
if torch_device == "mps":

0 commit comments

Comments
 (0)