SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Skip broken tests

Browse Source
This commit is contained in:
Sylvain Gugger
2022-08-11 09:33:41 -04:00
parent 4c8ec66a74
commit 3f0707b2fe
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/models/owlvit/test_modeling_owlvit.py
View File

@@ -733,8 +733,9 @@ def prepare_img():
@require_vision @require_vision
@require_torch @require_torch
@unittest.skip("These tests are broken, fix me Alara")
class OwlViTModelIntegrationTest(unittest.TestCase): class OwlViTModelIntegrationTest(unittest.TestCase):
# @slow @slow
def test_inference(self): def test_inference(self):
model_name = "google/owlvit-base-patch32" model_name = "google/owlvit-base-patch32"
model = OwlViTModel.from_pretrained(model_name).to(torch_device) model = OwlViTModel.from_pretrained(model_name).to(torch_device)