remove further type check

Author: lars20070

stubs/outlines/models/vllm_offline.pyi

@@ -1,7 +1,10 @@
-from typing import Any
+from typing import TYPE_CHECKING
 
 from outlines.models.base import Model
 
+if TYPE_CHECKING:
+    from vllm import LLM
+
 class VLLMOffline(Model): ...
 
-def from_vllm_offline(model: Any) -> VLLMOffline: ...
+def from_vllm_offline(model: LLM) -> VLLMOffline: ...

tests/models/test_outlines.py

@@ -179,9 +179,7 @@ def sglang_model() -> OutlinesModel:
 
 @pytest.fixture
 def vllm_model_offline() -> OutlinesModel:  # pragma: no cover
-    outlines_model = outlines.models.vllm_offline.from_vllm_offline(  # type: ignore
-        vllm.LLM('microsoft/Phi-3-mini-4k-instruct')  # type: ignore
-    )
+    outlines_model = outlines.models.vllm_offline.from_vllm_offline(vllm.LLM('microsoft/Phi-3-mini-4k-instruct'))
     return OutlinesModel(outlines_model, provider=OutlinesProvider())
 
 
@@ -226,7 +224,7 @@ def binary_image() -> BinaryImage:
     ),
     pytest.param(
         'from_vllm_offline',
-        lambda: (vllm.LLM('microsoft/Phi-3-mini-4k-instruct'),),  # type: ignore
+        lambda: (vllm.LLM('microsoft/Phi-3-mini-4k-instruct'),),
         marks=skip_if_vllm_imports_unsuccessful,
     ),
 ]
@@ -285,7 +283,7 @@ def test_init(model_loading_function_name: str, args: Callable[[], tuple[Any]])
     ),
     pytest.param(
         'from_vllm_offline',
-        lambda: (vllm.LLM('microsoft/Phi-3-mini-4k-instruct'),),  # type: ignore
+        lambda: (vllm.LLM('microsoft/Phi-3-mini-4k-instruct'),),
         marks=skip_if_vllm_imports_unsuccessful,
     ),
 ]