fix paths in pre commit configuration

.github/workflows/python_ci.yml

@@ -161,169 +161,169 @@ jobs:
             pip3 install outlines-core --find-links dist --force-reinstall
             pytest
 
-  musllinux:
-    runs-on: ${{ matrix.platform.runner }}
-    defaults:
-      run:
-        working-directory: bindings/python
-    strategy:
-      matrix:
-        platform:
-          - runner: ubuntu-latest
-            target: x86_64
-          - runner: ubuntu-latest
-            target: x86
-          - runner: ubuntu-latest
-            target: aarch64
-          - runner: ubuntu-latest
-            target: armv7
-    steps:
-      - uses: actions/checkout@v4
-      - uses: actions/setup-python@v5
-        with:
-          python-version: 3.x
-      - name: Build wheels
-        uses: PyO3/maturin-action@v1
-        with:
-          target: ${{ matrix.platform.target }}
-          args: --release --out dist --find-interpreter
-          sccache: 'true'
-          manylinux: musllinux_1_2
-          working-directory: bindings/python
-      - name: Upload wheels
-        uses: actions/upload-artifact@v4
-        with:
-          name: wheels-musllinux-${{ matrix.platform.target }}
-          path: bindings/python/dist
-      - name: pytest
-        if: ${{ startsWith(matrix.platform.target, 'x86_64') }}
-        uses: addnab/docker-run-action@v3
-        with:
-          image: alpine:latest
-          options: -v ${{ github.workspace }}:/io -w /io/bindings/python
-          run: |
-            set -e
-            apk add py3-pip py3-virtualenv
-            python3 -m virtualenv .venv
-            source .venv/bin/activate
-            pip install outlines-core --no-index --find-links dist --force-reinstall
-            pip install pytest
-            pytest
-      - name: pytest
-        if: ${{ !startsWith(matrix.platform.target, 'x86') }}
-        uses: uraimo/run-on-arch-action@v2
-        with:
-          arch: ${{ matrix.platform.target }}
-          distro: alpine_latest
-          githubToken: ${{ github.token }}
-          install: |
-            apk add py3-virtualenv
-          run: |
-            set -e
-            cd bindings/python
-            python3 -m virtualenv .venv
-            source .venv/bin/activate
-            pip install pytest
-            pip install outlines-core --find-links dist --force-reinstall
-            pytest
+#   musllinux:
+#     runs-on: ${{ matrix.platform.runner }}
+#     defaults:
+#       run:
+#         working-directory: bindings/python
+#     strategy:
+#       matrix:
+#         platform:
+#           - runner: ubuntu-latest
+#             target: x86_64
+#           - runner: ubuntu-latest
+#             target: x86
+#           - runner: ubuntu-latest
+#             target: aarch64
+#           - runner: ubuntu-latest
+#             target: armv7
+#     steps:
+#       - uses: actions/checkout@v4
+#       - uses: actions/setup-python@v5
+#         with:
+#           python-version: 3.x
+#       - name: Build wheels
+#         uses: PyO3/maturin-action@v1
+#         with:
+#           target: ${{ matrix.platform.target }}
+#           args: --release --out dist --find-interpreter
+#           sccache: 'true'
+#           manylinux: musllinux_1_2
+#           working-directory: bindings/python
+#       - name: Upload wheels
+#         uses: actions/upload-artifact@v4
+#         with:
+#           name: wheels-musllinux-${{ matrix.platform.target }}
+#           path: bindings/python/dist
+#       - name: pytest
+#         if: ${{ startsWith(matrix.platform.target, 'x86_64') }}
+#         uses: addnab/docker-run-action@v3
+#         with:
+#           image: alpine:latest
+#           options: -v ${{ github.workspace }}:/io -w /io/bindings/python
+#           run: |
+#             set -e
+#             apk add py3-pip py3-virtualenv
+#             python3 -m virtualenv .venv
+#             source .venv/bin/activate
+#             pip install outlines-core --no-index --find-links dist --force-reinstall
+#             pip install pytest
+#             pytest
+#       - name: pytest
+#         if: ${{ !startsWith(matrix.platform.target, 'x86') }}
+#         uses: uraimo/run-on-arch-action@v2
+#         with:
+#           arch: ${{ matrix.platform.target }}
+#           distro: alpine_latest
+#           githubToken: ${{ github.token }}
+#           install: |
+#             apk add py3-virtualenv
+#           run: |
+#             set -e
+#             cd bindings/python
+#             python3 -m virtualenv .venv
+#             source .venv/bin/activate
+#             pip install pytest
+#             pip install outlines-core --find-links dist --force-reinstall
+#             pytest
 
-  windows:
-    runs-on: ${{ matrix.platform.runner }}
-    defaults:
-      run:
-        working-directory: bindings/python
-    strategy:
-      matrix:
-        platform:
-          - runner: windows-latest
-            target: x64
-          - runner: windows-latest
-            target: x86
-    steps:
-      - uses: actions/checkout@v4
-      - uses: actions/setup-python@v5
-        with:
-          python-version: 3.x
-          architecture: ${{ matrix.platform.target }}
-      - name: Build wheels
-        uses: PyO3/maturin-action@v1
-        with:
-          target: ${{ matrix.platform.target }}
-          args: --release --out dist --find-interpreter
-          sccache: 'true'
-          working-directory: bindings/python
-      - name: Upload wheels
-        uses: actions/upload-artifact@v4
-        with:
-          name: wheels-windows-${{ matrix.platform.target }}
-          path: bindings/python/dist
-      - name: pytest
-        if: ${{ !startsWith(matrix.platform.target, 'aarch64') }}
-        shell: bash
-        run: |
-          set -e
-          python3 -m venv .venv
-          source .venv/Scripts/activate
-          pip install outlines-core --find-links dist --force-reinstall
-          pip install pytest
-          pytest
+#   windows:
+#     runs-on: ${{ matrix.platform.runner }}
+#     defaults:
+#       run:
+#         working-directory: bindings/python
+#     strategy:
+#       matrix:
+#         platform:
+#           - runner: windows-latest
+#             target: x64
+#           - runner: windows-latest
+#             target: x86
+#     steps:
+#       - uses: actions/checkout@v4
+#       - uses: actions/setup-python@v5
+#         with:
+#           python-version: 3.x
+#           architecture: ${{ matrix.platform.target }}
+#       - name: Build wheels
+#         uses: PyO3/maturin-action@v1
+#         with:
+#           target: ${{ matrix.platform.target }}
+#           args: --release --out dist --find-interpreter
+#           sccache: 'true'
+#           working-directory: bindings/python
+#       - name: Upload wheels
+#         uses: actions/upload-artifact@v4
+#         with:
+#           name: wheels-windows-${{ matrix.platform.target }}
+#           path: bindings/python/dist
+#       - name: pytest
+#         if: ${{ !startsWith(matrix.platform.target, 'aarch64') }}
+#         shell: bash
+#         run: |
+#           set -e
+#           python3 -m venv .venv
+#           source .venv/Scripts/activate
+#           pip install outlines-core --find-links dist --force-reinstall
+#           pip install pytest
+#           pytest
 
-  macos:
-    runs-on: ${{ matrix.platform.runner }}
-    defaults:
-      run:
-        working-directory: bindings/python
-    strategy:
-      matrix:
-        platform:
-          - runner: macos-12
-            target: x86_64
-          - runner: macos-14
-            target: aarch64
-    steps:
-      - uses: actions/checkout@v4
-      - uses: actions/setup-python@v5
-        with:
-          python-version: 3.x
-      - name: Build wheels
-        uses: PyO3/maturin-action@v1
-        with:
-          target: ${{ matrix.platform.target }}
-          args: --release --out dist --find-interpreter
-          sccache: 'true'
-          working-directory: bindings/python
-      - name: Upload wheels
-        uses: actions/upload-artifact@v4
-        with:
-          name: wheels-macos-${{ matrix.platform.target }}
-          path: bindings/python/dist
-      - name: pytest
-        run: |
-          set -e
-          python3 -m venv .venv
-          source .venv/bin/activate
-          pip install outlines-core --find-links dist --force-reinstall
-          pip install pytest
-          pytest
+#   macos:
+#     runs-on: ${{ matrix.platform.runner }}
+#     defaults:
+#       run:
+#         working-directory: bindings/python
+#     strategy:
+#       matrix:
+#         platform:
+#           - runner: macos-12
+#             target: x86_64
+#           - runner: macos-14
+#             target: aarch64
+#     steps:
+#       - uses: actions/checkout@v4
+#       - uses: actions/setup-python@v5
+#         with:
+#           python-version: 3.x
+#       - name: Build wheels
+#         uses: PyO3/maturin-action@v1
+#         with:
+#           target: ${{ matrix.platform.target }}
+#           args: --release --out dist --find-interpreter
+#           sccache: 'true'
+#           working-directory: bindings/python
+#       - name: Upload wheels
+#         uses: actions/upload-artifact@v4
+#         with:
+#           name: wheels-macos-${{ matrix.platform.target }}
+#           path: bindings/python/dist
+#       - name: pytest
+#         run: |
+#           set -e
+#           python3 -m venv .venv
+#           source .venv/bin/activate
+#           pip install outlines-core --find-links dist --force-reinstall
+#           pip install pytest
+#           pytest
 
-  sdist:
-    runs-on: ubuntu-latest
-    defaults:
-      run:
-        working-directory: bindings/python
-    steps:
-      - uses: actions/checkout@v4
-      - name: Build sdist
-        uses: PyO3/maturin-action@v1
-        with:
-          command: sdist
-          args: --out dist
-          working-directory: bindings/python
-      - name: Upload sdist
-        uses: actions/upload-artifact@v4
-        with:
-          name: wheels-sdist
-          path: bindings/python/dist
+#   sdist:
+#     runs-on: ubuntu-latest
+#     defaults:
+#       run:
+#         working-directory: bindings/python
+#     steps:
+#       - uses: actions/checkout@v4
+#       - name: Build sdist
+#         uses: PyO3/maturin-action@v1
+#         with:
+#           command: sdist
+#           args: --out dist
+#           working-directory: bindings/python
+#       - name: Upload sdist
+#         uses: actions/upload-artifact@v4
+#         with:
+#           name: wheels-sdist
+#           path: bindings/python/dist
 
 #   release:
 #     name: Release

.pre-commit-config.yaml

@@ -28,7 +28,7 @@ repos:
   rev: v1.2.0
   hooks:
     - id: mypy
-      args: [--allow-redefinition]
+      args: [--config-file=bindings/python/pyproject.toml, --allow-redefinition]
       exclude: ^examples/
       additional_dependencies: [types-tqdm, types-Pillow]
 - repo: local

bindings/python/outlines_core/__init__.py

@@ -1,5 +1,5 @@
 """Outlines is a Generative Model Programming Framework."""
-import outlines_core.models
-from _outlines_core_rs import __version__
+import outlines_core.models  # noqa: F401
+from _outlines_core_rs import __version__  # type: ignore
 
 __all__ = ["models", "__version__"]

bindings/python/outlines_core/models/__init__.py

@@ -8,6 +8,11 @@
 
 from typing import Union
 
-from .transformers import Transformers, TransformerTokenizer, mamba, transformers
+from .transformers import (  # noqa: F401
+    Transformers,
+    TransformerTokenizer,
+    mamba,
+    transformers,
+)
 
 LogitsGenerator = Union[Transformers]

bindings/python/pyproject.toml

@@ -101,6 +101,7 @@ module = [
     "setuptools_rust.*",
     # TODO: Add type info for the Rust extension
     "outlines_core.fsm.outlines_core_rs.*",
+    "_outlines_core_rs.*",
 ]
 ignore_missing_imports = true
 

setup.cfg

@@ -3,6 +3,6 @@ max-line-length = 88
 select = C,E,F,W
 ignore = E203,E231,E501,E741,W503,W504,C901,E731
 per-file-ignores =
-    **/__init__.py:F401,F403
+    bindings/python/outlines_core/__init__.py:F401,F403
 exclude =
-    outlines_core/_version.py
+    bindings/python/outlines_core/_version.py