BREAK: drop support for Python 3.9 (#552)

redeboer · web-flow · commit 23c8ff54fb41 · 2025-10-13T19:55:37.000+02:00
* ENH: set `zip(..., strict=True)`
diff --git a/benchmarks/ampform.py b/benchmarks/ampform.py
@@ -2,7 +2,7 @@
 
 from collections.abc import Sequence
 from pprint import pprint
-from typing import TYPE_CHECKING, Union
+from typing import TYPE_CHECKING
 
 import numpy as np
 import pytest
@@ -31,7 +31,7 @@
         ParametrizedFunction,
     )
 
-StateDefinition = Union[str, tuple[str, Sequence[float]]]
+StateDefinition = str | tuple[str, Sequence[float]]
 
 
 def formulate_amplitude_model(
diff --git a/docs/amplitude-analysis.ipynb b/docs/amplitude-analysis.ipynb
@@ -765,7 +765,7 @@
     "    density=True,\n",
     ")\n",
     "ax.set_xlabel(\"$m$ [GeV]\")\n",
-    "for p, color in zip(resonances, colors):\n",
+    "for p, color in zip(resonances, colors, strict=True):\n",
     "    ax.axvline(x=p.mass, linestyle=\"dotted\", label=p.name, color=color)\n",
     "ax.legend()\n",
     "plt.show()"
@@ -1476,7 +1476,7 @@
     "\n",
     "def indicate_masses(ax):\n",
     "    ax.set_xlabel(\"$m$ [GeV]\")\n",
-    "    for color, resonance in zip(colors, resonances):\n",
+    "    for color, resonance in zip(colors, resonances, strict=True):\n",
     "        ax.axvline(\n",
     "            x=resonance.mass,\n",
     "            linestyle=\"dotted\",\n",
diff --git a/docs/amplitude-analysis/analytic-continuation.ipynb b/docs/amplitude-analysis/analytic-continuation.ipynb
@@ -255,7 +255,7 @@
     "\n",
     "fig, ax = plt.subplots()\n",
     "ax.set_xlabel(\"$m_{02}$ [GeV]\")\n",
-    "for p, color in zip(resonances, colors):\n",
+    "for p, color in zip(resonances, colors, strict=True):\n",
     "    ax.axvline(x=p.mass, linestyle=\"dotted\", label=p.name, color=color)\n",
     "ax.set_yticks([])\n",
     "ax.hist(phsp[\"m_01\"], bins=100, alpha=0.5, weights=intensities)\n",
diff --git a/pyproject.toml b/pyproject.toml
@@ -19,7 +19,6 @@ classifiers = [
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
     "Programming Language :: Python :: 3.13",
-    "Programming Language :: Python :: 3.9",
     "Programming Language :: Python",
     "Topic :: Scientific/Engineering :: Physics",
     "Topic :: Scientific/Engineering",
@@ -32,7 +31,6 @@ dependencies = [
     "numpy",
     "sympy >=1.9", # lambdify cse
     "tqdm >=4.24.0", # autonotebook
-    'typing-extensions; python_version < "3.10"',
 ]
 description = "Python fitter package for multiple computational back-ends"
 dynamic = ["version"]
@@ -48,7 +46,7 @@ keywords = [
 license = {file = "LICENSE"}
 maintainers = [{email = "compwa-admin@ep1.rub.de"}]
 name = "tensorwaves"
-requires-python = ">=3.9"
+requires-python = ">=3.10"
 
 [project.optional-dependencies]
 jax = [
@@ -372,7 +370,6 @@ positional = true
 [tool.poe.tasks.test-all]
 help = "Run all tests on each supported Python version"
 sequence = [
-    {ref = "test-py 3.9"},
     {ref = "test-py 3.10"},
     {ref = "test-py 3.11"},
     {ref = "test-py 3.12"},
diff --git a/src/tensorwaves/data/_data_sample.py b/src/tensorwaves/data/_data_sample.py
@@ -3,11 +3,13 @@
 from __future__ import annotations
 
 import operator
-from typing import TYPE_CHECKING, Any, Callable
+from typing import TYPE_CHECKING, Any
 
 import numpy as np
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
+
     from tqdm.auto import tqdm
 
     from tensorwaves.interface import DataSample
diff --git a/src/tensorwaves/data/rng.py b/src/tensorwaves/data/rng.py
@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Optional, Union
+from typing import TYPE_CHECKING
 
 import numpy as np
 
@@ -12,7 +12,7 @@
 if TYPE_CHECKING:  # pragma: no cover
     import tensorflow as tf
 
-    SeedLike = Optional[Union[int, tf.random.Generator]]
+    SeedLike = int | tf.random.Generator | None
 
 
 class NumpyUniformRNG(RealNumberGenerator):
diff --git a/src/tensorwaves/estimator.py b/src/tensorwaves/estimator.py
@@ -5,7 +5,7 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Callable
+from typing import TYPE_CHECKING
 
 from tensorwaves.data.transform import SympyDataTransformer
 from tensorwaves.function._backend import find_function, raise_missing_module_error
@@ -19,7 +19,7 @@
 )
 
 if TYPE_CHECKING:
-    from collections.abc import Iterable, Mapping
+    from collections.abc import Callable, Iterable, Mapping
 
     import numpy as np
     import sympy as sp
diff --git a/src/tensorwaves/function/__init__.py b/src/tensorwaves/function/__init__.py
@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 import inspect
-from typing import TYPE_CHECKING, Callable
+from typing import TYPE_CHECKING
 
 import attrs
 import numpy as np
@@ -17,7 +17,7 @@
 )
 
 if TYPE_CHECKING:
-    from collections.abc import Iterable, Mapping
+    from collections.abc import Callable, Iterable, Mapping
 
 
 def _all_str(
diff --git a/src/tensorwaves/function/_backend.py b/src/tensorwaves/function/_backend.py
@@ -3,17 +3,12 @@
 from __future__ import annotations
 
 from functools import partial
-from typing import TYPE_CHECKING, Callable
+from typing import TYPE_CHECKING
 from warnings import warn
 
 if TYPE_CHECKING:
-    import sys
-    from typing import TypeVar
-
-    if sys.version_info >= (3, 10):
-        from typing import ParamSpec
-    else:
-        from typing_extensions import ParamSpec
+    from collections.abc import Callable
+    from typing import ParamSpec, TypeVar
 
     P = ParamSpec("P")
     T = TypeVar("T")
diff --git a/src/tensorwaves/function/sympy/__init__.py b/src/tensorwaves/function/sympy/__init__.py
@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 import logging
-from typing import TYPE_CHECKING, Any, Callable
+from typing import TYPE_CHECKING, Any
 
 from tqdm.auto import tqdm
 
@@ -15,7 +15,7 @@
 )
 
 if TYPE_CHECKING:  # pragma: no cover
-    from collections.abc import Generator, Iterable, Mapping, Sequence
+    from collections.abc import Callable, Generator, Iterable, Mapping, Sequence
 
     import sympy as sp
     from sympy.printing.printer import Printer
diff --git a/src/tensorwaves/function/sympy/_printer.py b/src/tensorwaves/function/sympy/_printer.py
@@ -1,12 +1,12 @@
 from __future__ import annotations
 
 import re
-from typing import TYPE_CHECKING, Any, Callable, TypeVar
+from typing import TYPE_CHECKING, Any, TypeVar
 
 from sympy.printing.numpy import NumPyPrinter
 
 if TYPE_CHECKING:  # pragma: no cover
-    from collections.abc import Iterable
+    from collections.abc import Callable, Iterable
 
     import sympy as sp
 
diff --git a/src/tensorwaves/interface.py b/src/tensorwaves/interface.py
@@ -4,7 +4,7 @@
 
 from abc import ABC, abstractmethod
 from collections.abc import Mapping
-from typing import TYPE_CHECKING, Any, Generic, TypeVar, Union
+from typing import TYPE_CHECKING, Any, Generic, TypeVar
 
 import attrs
 import numpy as np
@@ -38,7 +38,7 @@ def __call__(self, data: InputType) -> OutputType: ...
 
 DataSample = dict[str, np.ndarray]
 """Mapping of variable names to a sequence of data points, used by `Function`."""
-ParameterValue = Union[complex, float]
+ParameterValue = complex | float
 """Allowed types for parameter values."""
 
 
diff --git a/src/tensorwaves/optimizer/minuit.py b/src/tensorwaves/optimizer/minuit.py
@@ -5,7 +5,7 @@
 
 import logging
 import time
-from typing import TYPE_CHECKING, Any, Callable
+from typing import TYPE_CHECKING, Any
 
 import iminuit
 from tqdm.auto import tqdm
@@ -18,7 +18,7 @@
 from tensorwaves.optimizer.parameter import ParameterFlattener
 
 if TYPE_CHECKING:
-    from collections.abc import Iterable, Mapping
+    from collections.abc import Callable, Iterable, Mapping
 
 _LOGGER = logging.getLogger(__name__)
 
diff --git a/src/tensorwaves/optimizer/scipy.py b/src/tensorwaves/optimizer/scipy.py
@@ -79,7 +79,7 @@ def create_parameter_dict(
             pars: Iterable[float],
         ) -> dict[str, ParameterValue]:
             return parameter_handler.unflatten(
-                dict(zip(flattened_parameters.keys(), pars))
+                dict(zip(flattened_parameters.keys(), pars, strict=True))
             )
 
         def wrapped_function(pars: list) -> float:
diff --git a/tests/function/test_sympy.py b/tests/function/test_sympy.py
@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import logging
-from typing import TYPE_CHECKING, Callable
+from typing import TYPE_CHECKING
 
 import numpy as np
 import pytest
@@ -18,6 +18,8 @@
 )
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
+
     from _pytest.logging import LogCaptureFixture
 
 __symbols: tuple[sp.Symbol, ...] = sp.symbols("a b c d x y z")
diff --git a/tests/optimizer/test_minuit.py b/tests/optimizer/test_minuit.py
@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Callable
+from typing import TYPE_CHECKING
 
 import pytest
 
@@ -10,7 +10,7 @@
 from . import CallbackMock, assert_invocations
 
 if TYPE_CHECKING:
-    from collections.abc import Mapping
+    from collections.abc import Callable, Mapping
 
     from pytest_mock import MockerFixture
 
diff --git a/tests/optimizer/test_scipy.py b/tests/optimizer/test_scipy.py
@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Callable
+from typing import TYPE_CHECKING
 
 import pytest
 
@@ -10,7 +10,7 @@
 from . import CallbackMock, assert_invocations
 
 if TYPE_CHECKING:
-    from collections.abc import Mapping
+    from collections.abc import Callable, Mapping
 
     from pytest_mock import MockerFixture
 
diff --git a/uv.lock b/uv.lock
