Small fixes to re-enable a few PYI rules

Author: Avasam

pyproject.toml

@@ -56,12 +56,8 @@ ignore = [
 
     # TODO: Investigate and fix or configure
     "PYI001",
-    "PYI002",
-    "PYI017",
     "PYI024",
-    "PYI048",
     "PYI051", # Request for autofix: https://github.com/astral-sh/ruff/issues/14185
-    "PYI052",
 ]
 [tool.ruff.lint.per-file-ignores]
 "*.pyi" = [

stubs/sympy-stubs/codegen/ast.pyi

@@ -147,7 +147,9 @@ class FloatBaseType(Type):
 class FloatType(FloatBaseType):
     __slots__ = ...
     _fields = ...
-    _construct_nexp = _construct_nmant = _construct_nbits = Integer
+    _construct_nexp = Integer
+    _construct_nmant = Integer
+    _construct_nbits = Integer
     @property
     def max_exponent(self) -> Expr: ...
     @property

stubs/sympy-stubs/matrices/dense.pyi

@@ -21,7 +21,8 @@ class DenseMatrix(RepMatrix):
 class MutableDenseMatrix(DenseMatrix, MutableRepMatrix):
     def simplify(self, **kwargs) -> None: ...
 
-Matrix = MutableMatrix = MutableDenseMatrix
+Matrix = MutableDenseMatrix
+MutableMatrix = MutableDenseMatrix
 
 def list2numpy(l, dtype=...) -> NDArray[Any, Any]: ...
 def matrix2numpy(m, dtype=...) -> NDArray[Any, Any]: ...

stubs/sympy-stubs/polys/domains/simpledomain.pyi

@@ -3,5 +3,5 @@ from typing_extensions import Self
 from sympy.polys.domains.domain import Domain
 
 class SimpleDomain(Domain):
-    is_Simple = True
+    is_Simple: bool = True
     def inject(self, *gens) -> Self: ...

stubs/sympy-stubs/polys/polymatrix.pyi

@@ -39,4 +39,5 @@ class MutablePolyDenseMatrix:
     def nullspace(self) -> list[Self]: ...
     def rank(self): ...
 
-PolyMatrix = MutablePolyMatrix = MutablePolyDenseMatrix
+PolyMatrix = MutablePolyDenseMatrix
+MutablePolyMatrix = MutablePolyDenseMatrix

tests/run_tests.py

@@ -14,14 +14,14 @@ def install_requirements(test_folder: Path):
 def run_pyright():
     print("\nRunning Pyright...")
     # https://github.com/RobertCraigie/pyright-python#keeping-pyright-and-pylance-in-sync
-    del os.environ["PYRIGHT_PYTHON_FORCE_VERSION"]
+    os.environ.pop("PYRIGHT_PYTHON_FORCE_VERSION", None)
     os.environ["PYRIGHT_PYTHON_PYLANCE_VERSION"] = "latest-prerelease"
     return subprocess.run((sys.executable, "-m", "pyright"))
 
 
 def run_mypy():
     print("\nRunning mypy...")
-    return subprocess.run((sys.executable, "-m", "mypy"))
+    return subprocess.run((sys.executable, "-m", "mypy", "."))
 
 
 def main():