Get mypy-clean, enable mypy in CI

Also, required from __future__ import annotations

Author: inducer

.github/workflows/ci.yml

@@ -25,6 +25,23 @@ jobs:
         -   uses: actions/checkout@v4
         -   uses: crate-ci/typos@master
 
+    mypy:
+        name: Mypy
+        runs-on: ubuntu-latest
+        steps:
+        -   uses: actions/checkout@v4
+        -
+            uses: actions/setup-python@v5
+            with:
+                python-version: '3.x'
+        -   name: "Main Script"
+            run: |
+                curl -L -O https://tiker.net/ci-support-v0
+                . ./ci-support-v0
+                build_py_project_in_conda_env
+                python -m pip install mypy
+                mypy $(get_proj_name)
+
     pylint:
         name: Pylint
         runs-on: ubuntu-latest

.gitlab-ci.yml

@@ -140,6 +140,18 @@ Pylint:
   except:
   - tags
 
+Mypy:
+  script: |
+    curl -L -O https://tiker.net/ci-support-v0
+    . ./ci-support-v0
+    build_py_project_in_venv
+    python -m pip install mypy
+    mypy $(get_proj_name)
+  tags:
+  - python3
+  except:
+  - tags
+
 Downstream:
   parallel:
     matrix:

benchmarks/bench_translations.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 
 import numpy as np

pyproject.toml

@@ -16,6 +16,9 @@ extend-select = [
     "UP",  # pyupgrade
     "RUF", # ruff
     "W",   # pycodestyle
+
+    # TODO
+    # "SIM",
 ]
 extend-ignore = [
     "C90",  # McCabe complexity
@@ -42,6 +45,12 @@ known-local-folder = [
     "sumpy",
 ]
 lines-after-imports = 2
+required-imports = ["from __future__ import annotations"]
+
+[tool.ruff.lint.per-file-ignores]
+"doc/**/*.py" = ["I002"]
+"examples/**/*.py" = ["I002"]
+"setup.py" = ["I002"]
 
 [tool.typos.default]
 extend-ignore-re = [
@@ -61,3 +70,23 @@ extend-exclude = [
   "contrib/*/*.ipynb",
   "notes/*/*.eps",
 ]
+
+[tool.mypy]
+warn_unused_ignores = true
+
+[[tool.mypy.overrides]]
+module = [
+    "pyopencl.*",
+    "pymbolic.*",
+    "loopy.*",
+    "symengine.*",
+    "boxtree.*",
+    "pyvkfft.*",
+    "pyfmmlib.*",
+    "mayavi.*",
+    "scipy.*",
+    "sympy.*",
+    "matplotlib.*",
+    "pyvisfile.*",
+]
+ignore_missing_imports = true

sumpy/__init__.py

@@ -1,3 +1,6 @@
+from __future__ import annotations
+
+
 __copyright__ = "Copyright (C) 2013 Andreas Kloeckner"
 
 __license__ = """
@@ -21,7 +24,9 @@
 """
 
 import os
+from typing import Hashable
 
+import loopy as lp
 from pytools.persistent_dict import WriteOncePersistentDict
 
 from sumpy.e2e import (
@@ -56,8 +61,8 @@
 ]
 
 
-code_cache = WriteOncePersistentDict("sumpy-code-cache-v6-"+VERSION_TEXT,
-                                     safe_sync=False)
+code_cache: WriteOncePersistentDict[Hashable, lp.TranslationUnit] = \
+        WriteOncePersistentDict("sumpy-code-cache-v6-"+VERSION_TEXT, safe_sync=False)
 
 
 # {{{ optimization control

sumpy/array_context.py

@@ -1,3 +1,6 @@
+from __future__ import annotations
+
+
 __copyright__ = "Copyright (C) 2022 Alexandru Fikl"
 
 __license__ = """

sumpy/assignment_collection.py

@@ -1,3 +1,6 @@
+from __future__ import annotations
+
+
 __copyright__ = "Copyright (C) 2012 Andreas Kloeckner"
 
 __license__ = """

sumpy/codegen.py

@@ -1,3 +1,6 @@
+from __future__ import annotations
+
+
 __copyright__ = "Copyright (C) 2012 Andreas Kloeckner"
 
 __license__ = """

sumpy/cse.py

@@ -1,3 +1,6 @@
+from __future__ import annotations
+
+
 __copyright__ = """
 Copyright (C) 2017 Matt Wala
 Copyright (C) 2006-2016 SymPy Development Team

sumpy/derivative_taker.py

@@ -1,3 +1,6 @@
+from __future__ import annotations
+
+
 __copyright__ = """
 Copyright (C) 2012 Andreas Kloeckner
 Copyright (C) 2020 Isuru Fernando
@@ -338,7 +341,7 @@ def diff(self, mi, q=0):
 
 # {{{ DifferentiatedExprDerivativeTaker
 
-DerivativeCoeffDict = Dict[Tuple[int], Any]
+DerivativeCoeffDict = Dict[Tuple[int, ...], Any]
 
 
 @tag_dataclass
@@ -390,7 +393,7 @@ def diff_derivative_coeff_dict(derivative_coeff_dict: DerivativeCoeffDict,
     and return a new derivative transformation dictionary.
     """
     from collections import defaultdict
-    new_derivative_coeff_dict = defaultdict(lambda: 0)
+    new_derivative_coeff_dict: DerivativeCoeffDict = defaultdict(lambda: 0)
 
     for mi, coeff in derivative_coeff_dict.items():
         # In the case where we have x * u.diff(x), the result should