remove further type ignore

Author: lars20070

stubs/outlines/__init__.pyi

@@ -0,0 +1,3 @@
+from . import models
+
+__all__: list[str] = []

stubs/outlines/models/__init__.pyi

@@ -0,0 +1,3 @@
+from . import base, llamacpp, mlxlm, sglang, transformers, vllm_offline
+
+__all__: list[str] = []

stubs/outlines/models/llamacpp.pyi

@@ -0,0 +1,10 @@
+from typing import TYPE_CHECKING
+
+from outlines.models.base import Model
+
+if TYPE_CHECKING:
+    from llama_cpp import Llama
+
+class LlamaCpp(Model): ...
+
+def from_llamacpp(model: Llama) -> LlamaCpp: ...

stubs/outlines/models/sglang.pyi

@@ -0,0 +1,11 @@
+from typing import TYPE_CHECKING, Any, Union
+
+from outlines.models.base import AsyncModel, Model
+
+if TYPE_CHECKING:
+    from openai import AsyncOpenAI, OpenAI
+
+class SGLang(Model): ...
+class AsyncSGLang(AsyncModel): ...
+
+def from_sglang(client: OpenAI | AsyncOpenAI, *args: Any, **kwargs: Any) -> SGLang | AsyncSGLang: ...

tests/models/test_outlines.py

@@ -163,9 +163,7 @@ def llamacpp_model() -> OutlinesModel:
 
 @pytest.fixture
 def mlxlm_model() -> OutlinesModel:  # pragma: no cover
-    outlines_model = outlines.models.mlxlm.from_mlxlm(  # type: ignore[reportUnknownMemberType]
-        *mlx_lm.load('mlx-community/SmolLM-135M-Instruct-4bit')  # type: ignore
-    )
+    outlines_model = outlines.models.mlxlm.from_mlxlm(*mlx_lm.load('mlx-community/SmolLM-135M-Instruct-4bit'))
     return OutlinesModel(outlines_model, provider=OutlinesProvider())
 
 
@@ -214,7 +212,7 @@ def binary_image() -> BinaryImage:
     ),
     pytest.param(
         'from_mlxlm',
-        lambda: mlx_lm.load('mlx-community/SmolLM-135M-Instruct-4bit'),  # type: ignore
+        lambda: mlx_lm.load('mlx-community/SmolLM-135M-Instruct-4bit'),
         marks=skip_if_mlxlm_imports_unsuccessful,
     ),
     pytest.param(
@@ -273,7 +271,7 @@ def test_init(model_loading_function_name: str, args: Callable[[], tuple[Any]])
     ),
     pytest.param(
         'from_mlxlm',
-        lambda: mlx_lm.load('mlx-community/SmolLM-135M-Instruct-4bit'),  # type: ignore
+        lambda: mlx_lm.load('mlx-community/SmolLM-135M-Instruct-4bit'),
         marks=skip_if_mlxlm_imports_unsuccessful,
     ),
     pytest.param(