Draft

Author: RobinPicard

pydantic_ai_slim/pydantic_ai/models/outlines.py

@@ -1,6 +1,8 @@
 # There are linting and coverage escapes for MLXLM and VLLMOffline as the CI would not contain the right
 # environment to be able to run the associated tests
 
+# pyright: reportUnnecessaryTypeIgnoreComment = false
+
 from __future__ import annotations
 
 import io

tests/models/test_outlines.py

@@ -1,6 +1,8 @@
 # There are linting escapes for vllm offline as the CI would not contain the right
 # environment to load the associated dependencies
 
+# pyright: reportUnnecessaryTypeIgnoreComment = false
+
 from __future__ import annotations as _annotations
 
 import json
@@ -50,13 +52,13 @@
 with try_import() as llama_cpp_imports_successful:
     import llama_cpp
 
-with try_import() as vllm_imports_successful:  # pragma: no cover
+with try_import() as vllm_imports_successful:
     import vllm  # type: ignore[reportMissingImports]
 
     # We try to load the vllm model to ensure it is available
-    try:
+    try:  # pragma: no cover
         vllm.LLM('microsoft/Phi-3-mini-4k-instruct')  # type: ignore
-    except RuntimeError as e:
+    except RuntimeError as e:  # pragma: no cover
         if 'Found no NVIDIA driver' in str(e):
             # Treat as import failure
             raise ImportError('CUDA/NVIDIA driver not available') from e