Skip to content

Commit 0a0cfe3

Browse files
committed
fix on 3.10, 3.11
1 parent 5a50d88 commit 0a0cfe3

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/models/test_outlines.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -56,10 +56,10 @@
5656
import vllm # type: ignore[reportMissingImports]
5757

5858
# We try to load the vllm model to ensure it is available
59-
try: # pragma: no cover
59+
try: # pragma: no lax cover
6060
vllm.LLM('microsoft/Phi-3-mini-4k-instruct') # type: ignore
61-
except RuntimeError as e: # pragma: no cover
62-
if 'Found no NVIDIA driver' in str(e):
61+
except RuntimeError as e: # pragma: lax no cover
62+
if 'Found no NVIDIA driver' in str(e) or 'Device string must not be empty' in str(e):
6363
# Treat as import failure
6464
raise ImportError('CUDA/NVIDIA driver not available') from e
6565
raise

0 commit comments

Comments
 (0)