feat(gh-2110): setup for astral-sh/ty

.pre-commit-config.yaml

@@ -20,6 +20,14 @@ repos:
         entry: mypy --install-types --non-interactive
         files: ^numpyro/
 
+      - id: ty
+        name: ty check
+        language: python
+        entry: ty
+        args: [check, --config-file, ty.toml]
+        pass_filenames: false
+        always_run: true
+        additional_dependencies: [ty]
 
   - repo: https://github.com/pre-commit/pre-commit-hooks
     rev: v6.0.0
@@ -30,9 +38,9 @@ repos:
         exclude: notebooks/
 
   - repo: https://github.com/codespell-project/codespell
-    rev: v2.3.0
+    rev: v2.4.1
     hooks:
       - id: codespell
         stages: [pre-commit, commit-msg]
-        args:
-          [--ignore-words-list, "Teh,aas,ans,dout", --check-filenames, --skip, "*.ipynb"]
+        args: [--ignore-words-list, "Teh,aas,ans,dout", --check-filenames, --skip, "*.ipynb"]
+

Makefile

@@ -5,6 +5,7 @@ lint: FORCE
 	ruff format . --check
 	python scripts/update_headers.py --check
 	mypy --install-types --non-interactive numpyro
+	ty check -vvv --config-file ty.toml --exit-zero
 
 license: FORCE
 	python scripts/update_headers.py

pyproject.toml

@@ -130,6 +130,5 @@ module = [
     "numpyro.primitives.*",
     "numpyro.patch.*",
     "numpyro.util.*",
-    "numpyro.distributions.transforms",
 ]
 ignore_errors = false

setup.py

@@ -57,6 +57,7 @@
             "pyro-api>=0.1.1",
             "scikit-learn",
             "scipy>=1.9",
+            "ty>=0.0.4",
         ],
         "dev": [
             "dm-haiku>=0.0.14; python_version >= '3.10'",

ty.toml

@@ -0,0 +1,22 @@
+[environment]
+python-version = "3.12"
+
+[rules]
+unresolved-import = "ignore"
+
+[src]
+include = ["numpyro"]
+exclude = [
+    "numpyro.contrib.control_flow.*",
+    "numpyro.contrib.funsor.*",
+    "numpyro.contrib.hsgp.*",
+    "numpyro.contrib.stochastic_support.*",
+    "numpyro.diagnostics.*",
+    "numpyro.handlers.*",
+    "numpyro.infer.elbo.*",
+    "numpyro.optim.*",
+    "numpyro.primitives.*",
+    "numpyro.patch.*",
+    "numpyro.util.*",
+]
+respect-ignore-files = true