mypy: Fix all no-untyped-call

Author: Avasam

pyproject.toml

@@ -118,7 +118,6 @@ disallow_incomplete_defs = false
 disallow_untyped_defs = false
 # Allow dynamic typing in our own code
 warn_return_any = false # TODO
-disallow_untyped_calls = false # TODO
 check_untyped_defs = true
 # Suppressing errors
 disable_error_code = [

stubs/matplotlib/colors.pyi

@@ -128,24 +128,42 @@ class CenteredNorm(Normalize):
     def __call__(self, value, clip: bool = ...): ...
 
 def make_norm_from_scale(scale_cls, base_norm_cls=..., *, init=...): ...
-@make_norm_from_scale(FuncScale, init=lambda functions, vmin=None, vmax=None, clip=False: None)
-class FuncNorm(Normalize): ...
 
-@make_norm_from_scale(functools.partial(LogScale, nonpositive="mask"))
+class FuncNorm(Normalize):
+    def __init__(
+        self,
+        functions: tuple[Callable, Callable],
+        vmin: float | None = ...,
+        vmax: float | None = ...,
+        clip: bool = ...,
+    ) -> None: ...
+
 class LogNorm(Normalize): ...
 
-@make_norm_from_scale(
-    SymmetricalLogScale,
-    init=lambda linthresh, linscale=1, vmin=None, vmax=None, clip=False, *, base=10: None,
-)
 class SymLogNorm(Normalize):
+    def __init__(
+        self,
+        linthresh: float,
+        linscale: float = ...,
+        vmin: float | None = ...,
+        vmax: float | None = ...,
+        clip: bool = ...,
+        *,
+        base: float = ...,
+    ) -> None: ...
     @property
     def linthresh(self): ...
     @linthresh.setter
     def linthresh(self, value): ...
 
-@make_norm_from_scale(AsinhScale, init=lambda linear_width=1, vmin=None, vmax=None, clip=False: None)
 class AsinhNorm(Normalize):
+    def __init__(
+        self,
+        linear_width: float = ...,
+        vmin: float | None = ...,
+        vmax: float | None = ...,
+        clip: bool = ...,
+    ) -> None: ...
     @property
     def linear_width(self): ...
     @linear_width.setter

stubs/matplotlib/transforms.pyi

@@ -199,7 +199,7 @@ class Transform(TransformNode):
     output_dims = ...
     is_separable = ...
     has_inverse = ...
-    def __init_subclass__(cls): ...
+    def __init_subclass__(cls) -> None: ...
     def __add__(self, other: Transform) -> Transform: ...
     @property
     def depth(self) -> int: ...

stubs/sympy-stubs/assumptions/assume.pyi

@@ -1,7 +1,7 @@
 from collections.abc import Generator
 from contextlib import contextmanager
 from typing import Any, Callable
-from typing_extensions import Self
+from typing_extensions import Never, Self
 
 from sympy.core.basic import Basic
 from sympy.logic.boolalg import Boolean
@@ -35,9 +35,9 @@ class Predicate(Boolean, metaclass=PredicateMeta):
     @property
     def name(self) -> str: ...
     @classmethod
-    def register(cls, *types, **kwargs): ...
+    def register(cls, *types: type, **kwargs: object) -> Callable[..., None]: ...
     @classmethod
-    def register_many(cls, *types, **kwargs) -> Callable[..., None]: ...
+    def register_many(cls, *types: type, **kwargs: object) -> Callable[..., None]: ...
     def __call__(self, *args) -> AppliedPredicate: ...
     def eval(self, args, assumptions=...) -> None: ...
 

tests/run_hygiene.py

@@ -4,22 +4,22 @@
 from pathlib import Path
 
 
-def install_requirements():
+def install_requirements() -> subprocess.CompletedProcess[bytes]:
     print("\nInstalling requirements...")
     return subprocess.run((sys.executable, "-m", "pip", "install", "--upgrade", "ruff"))
 
 
-def run_ruff_fix():
+def run_ruff_fix() -> subprocess.CompletedProcess[bytes]:
     print("\nRunning Ruff check --fix...")
     return subprocess.run((sys.executable, "-m", "ruff", "check", "--fix"))
 
 
-def run_ruff_format():
+def run_ruff_format() -> subprocess.CompletedProcess[bytes]:
     print("\nRunning Ruff format...")
     return subprocess.run((sys.executable, "-m", "ruff", "format"))
 
 
-def main():
+def main() -> None:
     test_folder = Path(__file__).parent
     root = test_folder.parent
     os.chdir(root)

tests/run_tests.py

@@ -4,27 +4,27 @@
 from pathlib import Path
 
 
-def install_requirements(test_folder: Path):
+def install_requirements(test_folder: Path) -> subprocess.CompletedProcess[bytes]:
     print("\nInstalling requirements...")
     return subprocess.run(
         (sys.executable, "-m", "pip", "install", "--upgrade", "-r", os.path.join(test_folder, "requirements.txt"))
     )
 
 
-def run_pyright():
+def run_pyright() -> subprocess.CompletedProcess[bytes]:
     print("\nRunning Pyright...")
     # https://github.com/RobertCraigie/pyright-python#keeping-pyright-and-pylance-in-sync
     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():
+def run_mypy() -> subprocess.CompletedProcess[bytes]:
     print("\nRunning mypy...")
     return subprocess.run((sys.executable, "-m", "mypy", "."))
 
 
-def main():
+def main() -> None:
     test_folder = Path(__file__).parent
     root = test_folder.parent
     os.chdir(root)

utils/count_ids.py

@@ -1,4 +1,5 @@
 #!/bin/python
+from __future__ import annotations
 
 __doc__ = """Count IDs.
 
@@ -23,7 +24,7 @@
 import docopt
 
 
-def count(root, suffix, regex, uniq):
+def count(root: str | None, suffix: str | None, regex: str | re.Pattern[str] | None, uniq: bool) -> None:
     if root is None:
         root = "."
     filepat = "*" if suffix is None else "*." + suffix[suffix.find(".") + 1 :]

utils/validate_stubs.py

@@ -2,21 +2,6 @@
 # ruff: noqa: F841 # TODO: plenty of unused variables. Validate what was the intent.
 from __future__ import annotations
 
-__doc__ = """Validate Stubs.
-
-Usage:
-  validate_stubs <package> [--path=<stubpath>] [--class=<c>] [--function=<f>]
-  validate_stubs -h | --help
-  validate_stubs --version
-
-Options:
-  -h --help          Show this screen.
-  --version          Show version.
-  --path=<stubpath>  Where to find stubs (default to parent directory)
-  --function=<f>     Restrict to the named function (or method if used with --class).
-  --class=<c>        Restrict to the named class.
-"""
-
 import importlib
 import importlib.machinery
 import inspect
@@ -32,6 +17,21 @@
 
 import docopt
 
+__doc__ = """Validate Stubs.
+
+Usage:
+  validate_stubs <package> [--path=<stubpath>] [--class=<c>] [--function=<f>]
+  validate_stubs -h | --help
+  validate_stubs --version
+
+Options:
+  -h --help          Show this screen.
+  --version          Show version.
+  --path=<stubpath>  Where to find stubs (default to parent directory)
+  --function=<f>     Restrict to the named function (or method if used with --class).
+  --class=<c>        Restrict to the named class.
+"""
+
 overloads: dict[str, Callable] = {}
 
 
@@ -66,7 +66,7 @@ def import_dual(m: str, stub_path: str) -> tuple:
 
     """
 
-    def _clean(m):
+    def _clean(m: str) -> None:
         to_del = [k for k in sys.modules.keys() if k == m or k.startswith(m + ".")]
         for k in to_del:
             del sys.modules[k]
@@ -99,7 +99,7 @@ class ItemType(Enum):
 
     def __init__(
         self, file: str, module: str, name: str, object_: object, type_: ItemType, children: dict[str, Item] | None = None
-    ):
+    ) -> None:
         self.file = file
         self.module = module
         self.name = name
@@ -109,25 +109,25 @@ def __init__(
         self.done = False
         self.analog: Item | None = None
 
-    def ismodule(self):
+    def ismodule(self) -> bool:
         return self.type_ == Item.ItemType.MODULE
 
-    def isclass(self):
+    def isclass(self) -> bool:
         return self.type_ == Item.ItemType.CLASS
 
-    def isfunction(self):
+    def isfunction(self) -> bool:
         return self.type_ == Item.ItemType.FUNCTION
 
     @staticmethod
-    def make_function(file: str, module: str, name: str, object_: object):
+    def make_function(file: str, module: str, name: str, object_: object) -> Item:
         return Item(file, module, name, object_, Item.ItemType.FUNCTION)
 
     @staticmethod
-    def make_class(file: str, module: str, name: str, object_: object, children: dict[str, Item]):
+    def make_class(file: str, module: str, name: str, object_: object, children: dict[str, Item]) -> Item:
         return Item(file, module, name, object_, Item.ItemType.CLASS, children)
 
     @staticmethod
-    def make_module(file: str, module: str, name: str, object_: object, children: dict[str, Item]):
+    def make_module(file: str, module: str, name: str, object_: object, children: dict[str, Item]) -> Item:
         return Item(file, module, name, object_, Item.ItemType.MODULE, children)