TMP: add mypy

idiap · Nov 20, 2024 · 4b68740 · 4b68740
1 parent a85854d
commit 4b68740
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -56,11 +56,14 @@ dev = [
     "pytest>=8",
     "ruff==0.6.9",
 ]
-# Dependencies for running the tests
 test = [
     "accelerate>=0.20.0",
     "torchvision>=0.15.1",
 ]
+mypy = [
+    "mypy>=1.13.0",
+    "types-psutil>=6.1.0.20241102",
+]
 
 [tool.uv]
 default-groups = ["dev", "test"]