Skip to content

Commit af40004

Browse files
authored
[Tests] proper skipping of request caching test (huggingface#8908)
proper skipping of request caching test
1 parent 5802c2e commit af40004

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

tests/models/test_modeling_common.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -124,11 +124,8 @@ 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.skipIf(torch_device == "mps", reason="Test not supported for MPS.")
127128
def test_one_request_upon_cached(self):
128-
# TODO: For some reason this test fails on MPS where no HEAD call is made.
129-
if torch_device == "mps":
130-
return
131-
132129
use_safetensors = False
133130

134131
with tempfile.TemporaryDirectory() as tmpdirname:

0 commit comments

Comments
 (0)