Skip to content

GitLab

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

[Tests] proper skipping of request caching test (#8908) 

proper skipping of request caching test
parent 5802c2e3
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 4 deletions
tests/models/test_modeling_common.py
View file @ af400040
...@@ -124,11 +124,8 @@ class ModelUtilsTest(unittest.TestCase): ...@@ -124,11 +124,8 @@ 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.skipIf(torch_device == "mps", reason="Test not supported for MPS.")
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.
if torch_device == "mps":
return
use_safetensors = False use_safetensors = False
with tempfile.TemporaryDirectory() as tmpdirname: with tempfile.TemporaryDirectory() as tmpdirname:
......
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