OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Skip test if torchvision is not available (#149494)

Browse Source 
The test unconditionally imports torchvision and fails if the isn't installed.
Skip it in this case.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/149494
Approved by: https://github.com/janeyx99
This commit is contained in:
Alexander Grund 2025-03-21 18:57:10 +00:00 committed by PyTorch MergeBot
parent b0a5d55c58
commit 19b763def1
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
test/inductor/test_benchmark_fusion.py
View File

@ -58,7 +58,10 @@ class BenchmarkFusionTestTemplate:
@slowTest @slowTest
def test_resnet18(self): def test_resnet18(self):
import torchvision try:
import torchvision
except ImportError:
self.skipTest("TorchVision not available")
model = torchvision.models.resnet18() model = torchvision.models.resnet18()
model.eval() model.eval()