Nothing to check for mypy

.pre-commit-config.yaml

@@ -24,16 +24,6 @@ repos:
   rev: 23.3.0
   hooks:
     - id: black
-- repo: https://github.com/pre-commit/mirrors-mypy
-  rev: v1.7.0
-  hooks:
-    - id: mypy
-      exclude: |
-          (?x)^(
-              tests/|
-              benchmarks/|
-              \.github/
-          )
 - repo: local
   hooks:
     - id: cargo-fmt

pyproject.toml

@@ -63,16 +63,6 @@ addopts = [
     "--import-mode=importlib"
 ]
 
-[tool.mypy]
-exclude=["tests", "benchmarks", ".github"]
-namespace_packages = true
-explicit_package_bases = true
-
-[[tool.mypy.overrides]]
-module = [
-    "pydantic.*",
-    "pytest",
-]
 ignore_missing_imports = true
 
 [tool.coverage.run]

tests/test_guide.py

@@ -1,6 +1,5 @@
 import copy
 import pickle
-from typing import Dict, List, Union
 
 import pytest
 from outlines_core import Guide, Index, Vocabulary
@@ -9,8 +8,7 @@
 @pytest.fixture(scope="session")
 def index() -> Index:
     eos_token_id = 3
-    # types here only to please mypy checks
-    tokens: Dict[Union[str, bytes], List[int]] = {"1": [1], "2": [2]}
+    tokens = {"1": [1], "2": [2]}
     regex = r"[1-9]"
 
     vocabulary = Vocabulary(eos_token_id, tokens)

tests/test_index.py

@@ -1,7 +1,6 @@
 import copy
 import gc
 import pickle
-from typing import Dict, List, Union
 
 import pytest
 from outlines_core import Index, Vocabulary
@@ -10,8 +9,7 @@
 @pytest.fixture(scope="session")
 def index() -> Index:
     eos_token_id = 3
-    # types here only to please mypy checks
-    tokens: Dict[Union[str, bytes], List[int]] = {"1": [1], "2": [2]}
+    tokens = {"1": [1], "2": [2]}
     regex = r"[1-9]"
 
     vocabulary = Vocabulary(eos_token_id, tokens)