add stubs to ruff config

Author: lars20070

pyproject.toml

@@ -142,6 +142,7 @@ include = [
     "clai/**/*.py",
     "tests/**/*.py",
     "docs/**/*.py",
+    "stubs/**/*.pyi",
 ]
 
 [tool.ruff.lint]
@@ -186,6 +187,7 @@ quote-style = "single"
 "examples/**/*.py" = ["D101", "D103"]
 "tests/**/*.py" = ["D"]
 "docs/**/*.py" = ["D"]
+"stubs/**/*.pyi" = ["F401"]
 
 [tool.pyright]
 stubPath = "stubs"

stubs/mlx/nn.pyi

@@ -1,5 +1,3 @@
 from typing import Any
 
-
-class Module:
-    ...
+class Module: ...

stubs/outlines/models/base.pyi

@@ -1,11 +1,10 @@
-from typing import Any, AsyncIterable, Iterable
-
+from collections.abc import AsyncIterable, Iterable
+from typing import Any
 
 class Model:
     def __call__(self, *args: Any, **kwargs: Any) -> Any: ...
     def stream(self, *args: Any, **kwargs: Any) -> Iterable[Any]: ...
 
-
 class AsyncModel(Model):
     async def __call__(self, *args: Any, **kwargs: Any) -> Any: ...
     def stream(self, *args: Any, **kwargs: Any) -> AsyncIterable[Any]: ...

stubs/outlines/models/mlxlm.pyi

@@ -1,12 +1,10 @@
 from typing import Any
-from outlines.models.base import Model
 
 from mlx.nn import Module
+from outlines.models.base import Model
 from transformers.tokenization_utils import PreTrainedTokenizer
 
-
 class MLXLM(Model):
     def __init__(self, *args: Any, **kwargs: Any) -> None: ...
 
-
 def from_mlxlm(model: Module, tokenizer: PreTrainedTokenizer) -> MLXLM: ...

stubs/outlines/models/transformers.pyi

@@ -1,23 +1,16 @@
 from typing import Any
-from outlines.models.base import Model
 
+from outlines.models.base import Model
 from transformers.modeling_utils import PreTrainedModel
 from transformers.processing_utils import ProcessorMixin
 from transformers.tokenization_utils import PreTrainedTokenizer
 
-
-class Transformers(Model):
-    ...
-
-
-class TransformersMultiModal(Model):
-    ...
-
+class Transformers(Model): ...
+class TransformersMultiModal(Model): ...
 
 def from_transformers(
     model: PreTrainedModel,
     tokenizer_or_processor: PreTrainedTokenizer | ProcessorMixin,
     *,
     device_dtype: Any = None,
 ) -> Transformers | TransformersMultiModal: ...
-

stubs/outlines/models/vllm_offline.pyi

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

stubs/transformers/__init__.pyi

@@ -5,25 +5,20 @@ from .modeling_utils import PreTrainedModel
 from .processing_utils import ProcessorMixin
 from .tokenization_utils import PreTrainedTokenizer
 
-
 class AutoModelForCausalLM(PreTrainedModel):
     @classmethod
     def from_pretrained(cls, *args: Any, **kwargs: Any) -> PreTrainedModel: ...
 
-
 class AutoTokenizer(PreTrainedTokenizer):
     @classmethod
     def from_pretrained(cls, *args: Any, **kwargs: Any) -> PreTrainedTokenizer: ...
 
-
 class AutoProcessor(ProcessorMixin):
     @classmethod
     def from_pretrained(cls, *args: Any, **kwargs: Any) -> ProcessorMixin: ...
 
-
 class LlavaForConditionalGeneration(PreTrainedModel):
     @classmethod
     def from_pretrained(cls, *args: Any, **kwargs: Any) -> PreTrainedModel: ...
 
-
 def from_pretrained(*args: Any, **kwargs: Any) -> Any: ...

stubs/transformers/modeling_utils.pyi

@@ -1,5 +1,3 @@
 from typing import Any
 
-
-class PreTrainedModel:
-    ...
+class PreTrainedModel: ...

stubs/transformers/processing_utils.pyi

@@ -1,5 +1,3 @@
 from typing import Any
 
-
-class ProcessorMixin:
-    ...
+class ProcessorMixin: ...

stubs/transformers/tokenization_utils.pyi

@@ -1,11 +1,8 @@
 from typing import Any
 
-
 class PreTrainedTokenizer:
     chat_template: str | None
 
     def __init__(self, *args: Any, **kwargs: Any) -> None: ...
 
-
-class ProcessorMixin:
-    ...
+class ProcessorMixin: ...