"vscode:/vscode.git/clone" did not exist on "ee3d69447f919703cebff4b17c1949621786fe04"
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):
if p1.data.ne(p2.data).sum() > 0:
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.")
def test_one_request_upon_cached(self):
use_safetensors = False
......
......@@ -146,6 +146,7 @@ class CustomPipeline(DiffusionPipeline):
class DownloadTests(unittest.TestCase):
@unittest.skip("Flaky behaviour on CI. Re-enable after migrating to new runners")
def test_one_request_upon_cached(self):
# TODO: For some reason this test fails on MPS where no HEAD call is made.
if torch_device == "mps":
......@@ -191,6 +192,7 @@ class DownloadTests(unittest.TestCase):
assert "scheduler" 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):
# TODO: For some reason this test fails on MPS where no HEAD call is made.
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