[feature] adding mypy for static typing

.pre-commit-config.yaml

@@ -27,7 +27,15 @@ repos:
         language: system
         types_or: [python, pyi, jupyter]
         require_serial: true
+      - id: ty
+        name: ty
+        description: "Run 'ty' for Python type checking"
+        entry: ty check
+        language: system
+        types: [python]
+        require_serial: true
+        pass_filenames: false
 
 # pre-commit.ci SaaS does not have our Nix environment so we skip Nix-packaged hooks
 ci:
-  skip: [nixfmt, uv-lock, ruff, ruff-format]
+  skip: [nixfmt, uv-lock, ruff, ruff-format, ty]

.vscode/extensions.json

@@ -1,12 +1,12 @@
 {
     "recommendations": [
         "myriad-dreamin.tinymist",
-        "ms-python.black-formatter",
         "charliermarsh.ruff",
         "mkhl.direnv",
         "editorconfig.editorconfig",
         "timonwong.shellcheck",
         "melak47.ninja-syntax",
-        "semgrep.semgrep"
+        "semgrep.semgrep",
+        "astral-sh.ty"
     ]
 }

.vscode/settings.json

@@ -1,19 +1,16 @@
 {
-    "python.analysis.typeCheckingMode": "basic",
-    "python.analysis.autoImportCompletions": true,
-    "[python]": {
-        "editor.formatOnSave": true,
-        "editor.codeActionsOnSave": {
-            "source.fixAll": "explicit",
-            "source.organizeImports": "explicit"
-        },
-        "editor.defaultFormatter": "charliermarsh.ruff"
-    },
+    // disable python language server since we're using astral-sh.ty
+    "python.languageServer": "None",
     "notebook.formatOnSave.enabled": true,
     "notebook.codeActionsOnSave": {
         "notebook.source.fixAll": "explicit",
         "notebook.source.organizeImports": "explicit"
     },
     "tinymist.formatterMode": "typstyle",
-    "tinymist.systemFonts": false
+    "tinymist.systemFonts": false,
+    // Format Python files on save using ruff
+    "[python]": {
+        "editor.formatOnSave": true,
+        "editor.defaultFormatter": "charliermarsh.ruff"
+    }
 }

README.md

@@ -5,6 +5,7 @@ A template repository implementing Palisade's coding guide. Features:
 - Nix + uv + direnv environment for development and CI
 - Figures and the paper build in CI
 - Lint and format pre-commit hooks
+- Python type checking with ty
 - VSCode settings and extensions
 
 ## Development
@@ -76,7 +77,7 @@ Use Cursor to edit code. It will suggest extensions to install when you open the
 ```bash
 cd template # automatically loads environment
 ninja # builds figures and the paper
-git commit # checks format and lints
+git commit # checks format, lints, and type checks
 ```
 
 ### Files to know

flake.nix

@@ -23,7 +23,6 @@
             ninja
             # Python
             uv
-            ruff
             # Typesetting
             typst
             texlive.combined.scheme-medium

pyproject.toml

@@ -8,6 +8,11 @@ dependencies = [
     "matplotlib>=3.10.0",
     "numpy>=2.2.2",
 ]
+[dependency-groups]
+dev = [
+    "ty>=0.0.1a18",
+    "ruff>=0.12.5",
+]
 
 [tool.pyright]
 # disable pyright in favor of ruff