Scope pyright to src/ only, fix unused import in test_jax

Test files use beartype/shapix patterns that produce expected pyright
false positives — exclude them from type checking.

Author: acecchini

pyproject.toml

@@ -35,7 +35,7 @@ python-preference = "only-managed"
 
 [tool.pyright]
 typeCheckingMode = "standard"
-include = ["src", "tests"]
+include = ["src"]
 reportUnusedImport = "warning"
 reportInvalidTypeForm = false
 pythonVersion = "3.12"