feat: migrate from mypy to ty

.github/workflows/code-quality.yml

@@ -36,7 +36,7 @@ jobs:
     steps:
       - uses: actions/checkout@v4
       - uses: ./.github/actions/setup
-      - run: make mypy
+      - run: make ty
 
   test:
     runs-on: ubuntu-latest

.gitignore

@@ -8,7 +8,7 @@
 __pycache__
 
 .venv
-.mypy_cache
+.ty_cache
 .ruff_cache
 .python-version
 

Makefile

@@ -8,8 +8,8 @@ lint:
 check_formatting:
 	uv run ruff format --check src/
 
-mypy:
-	uv run mypy src/ --exclude "src/research"
+ty:
+	uv run ty check src/ --exclude "src/research/**"
 
 test:
 	uv run pytest tests/ -v

docs/installation.md

@@ -34,5 +34,5 @@ make test
 Run type checking:
 
 ```bash
-make mypy
+make ty
 ```

docs/usage.md

@@ -39,7 +39,7 @@ make full_pipe  # Cleans, generates, and analyzes
 ```bash
 make format     # Format code with ruff
 make lint       # Check for linting issues
-make mypy       # Type checking
+make ty         # Type checking
 ```
 
 ### Testing

pyproject.toml

@@ -22,18 +22,18 @@ dependencies = [
     "dask>=2025.1.0",
     "ipykernel>=6.29.5",
     "matplotlib>=3.10.0",
-    "mypy>=1.15.0",
     "nltk>=3.9.1",
     "numpy>=2.2.2",
     "optuna>=4.4.0",
     "pandas>=2.2.3",
     "pydantic>=2.10.6",
+    "ruff>=0.9.3",
     "scipy>=1.15.2",
     "sentence-transformers>=4.1.0",
     "stubs>=1.0.0",
-    "ruff>=0.9.3",
     "torch>=2.6.0",
     "transformers>=4.48.3",
+    "ty>=0.0.4",
 ]
 
 [project.optional-dependencies]
@@ -131,7 +131,6 @@ ignore = [
     "UP007",  # Conflicts with pydantic
     "E501",   # Line length
     "COM812", # Conflicts with formatter
-    "D407",   # Conflicts with mkdocs
     "ANN401", # **kwargs and *args must be arbitrary type
     "D100",   # required docs
     "D101",
@@ -154,8 +153,5 @@ ignore = [
     "PLR2004", # Magic value comparisons (acceptable in tests)
 ]
 
-[tool.mypy]
-mypy_path = "src"
-strict = true
-python_version = "3.12"
-plugins = ["pydantic.mypy"]
+[tool.ty]
+

src/models/softmax_bottleneck_model.py

@@ -20,8 +20,8 @@ def __init__(
         weight_init: WeightInit | None = None,
     ) -> None:
         super().__init__()
-        self.hidden_dim = hidden_size
-        self.output_dim = output_size
+        self.hidden_dim: int = hidden_size
+        self.output_dim: int = output_size
         self.fc_in = nn.Linear(output_size, hidden_size)
         self.hidden_layers: list[nn.Linear] = []
         if activation is None:

src/recovery/utils.py

@@ -38,8 +38,8 @@ def __init__(
         lambda_norm: float = 1.0e2,
     ) -> None:
         super().__init__()
-        self.lambda_nonneg = lambda_nonneg
-        self.lambda_norm = lambda_norm
+        self.lambda_nonneg: float = lambda_nonneg
+        self.lambda_norm: float = lambda_norm
 
     def forward(
         self,