Skip to content

Commit 5d4a99c

Browse files
nnshah1mc-nv
authored andcommitted
updating to skip torch tests if torch.cuda not available
1 parent d4fab30 commit 5d4a99c

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

python/test/test_api.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,9 @@
4444

4545
try:
4646
import torch
47+
48+
if not torch.cuda.is_available():
49+
torch = None
4750
except ImportError:
4851
torch = None
4952

@@ -343,7 +346,7 @@ def test_ready(self):
343346
self.assertTrue(server.ready())
344347

345348
@pytest.mark.xfail(
346-
tritonserver.__version__ <= "2.42.0",
349+
tritonserver.__version__ <= "2.43.0",
347350
reason="Known issue on stop: Exit timeout expired. Exiting immediately",
348351
raises=tritonserver.InternalError,
349352
)

0 commit comments

Comments
 (0)