Unverified Commit 93983b67 authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

[CI] Skip flaky download tests in PR CI (#8945)

update
parent 41b705f4
...@@ -124,6 +124,7 @@ class ModelUtilsTest(unittest.TestCase): ...@@ -124,6 +124,7 @@ class ModelUtilsTest(unittest.TestCase):
if p1.data.ne(p2.data).sum() > 0: if p1.data.ne(p2.data).sum() > 0:
assert False, "Parameters not the same!" assert False, "Parameters not the same!"
@unittest.skip("Flaky behaviour on CI. Re-enable after migrating to new runners")
@unittest.skipIf(torch_device == "mps", reason="Test not supported for MPS.") @unittest.skipIf(torch_device == "mps", reason="Test not supported for MPS.")
def test_one_request_upon_cached(self): def test_one_request_upon_cached(self):
use_safetensors = False use_safetensors = False
......
...@@ -146,6 +146,7 @@ class CustomPipeline(DiffusionPipeline): ...@@ -146,6 +146,7 @@ class CustomPipeline(DiffusionPipeline):
class DownloadTests(unittest.TestCase): class DownloadTests(unittest.TestCase):
@unittest.skip("Flaky behaviour on CI. Re-enable after migrating to new runners")
def test_one_request_upon_cached(self): def test_one_request_upon_cached(self):
# TODO: For some reason this test fails on MPS where no HEAD call is made. # TODO: For some reason this test fails on MPS where no HEAD call is made.
if torch_device == "mps": if torch_device == "mps":
...@@ -191,6 +192,7 @@ class DownloadTests(unittest.TestCase): ...@@ -191,6 +192,7 @@ class DownloadTests(unittest.TestCase):
assert "scheduler" in os.listdir(cached_folder) assert "scheduler" in os.listdir(cached_folder)
assert "feature_extractor" in os.listdir(cached_folder) assert "feature_extractor" in os.listdir(cached_folder)
@unittest.skip("Flaky behaviour on CI. Re-enable after migrating to new runners")
def test_less_downloads_passed_object_calls(self): def test_less_downloads_passed_object_calls(self):
# TODO: For some reason this test fails on MPS where no HEAD call is made. # TODO: For some reason this test fails on MPS where no HEAD call is made.
if torch_device == "mps": if torch_device == "mps":
......
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