We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5a50d88 commit 0a0cfe3Copy full SHA for 0a0cfe3
tests/models/test_outlines.py
@@ -56,10 +56,10 @@
56
import vllm # type: ignore[reportMissingImports]
57
58
# We try to load the vllm model to ensure it is available
59
- try: # pragma: no cover
+ try: # pragma: no lax cover
60
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):
+ except RuntimeError as e: # pragma: lax no cover
+ if 'Found no NVIDIA driver' in str(e) or 'Device string must not be empty' in str(e):
63
# Treat as import failure
64
raise ImportError('CUDA/NVIDIA driver not available') from e
65
raise
0 commit comments