chore: dropped support for python 3.9

Author: tm91236

.github/ISSUE_TEMPLATE/bug_report.yml

@@ -41,7 +41,6 @@ body:
       description: Which version or versions of Python are you seeing the problem on?
       multiple: true
       options:
-        - "3.9"
         - "3.10"
         - "3.11"
         - "3.12"

.github/workflows/release.yml

@@ -45,7 +45,6 @@ jobs:
       fail-fast: false
       matrix:
         python-version:
-          - "3.9"
           - "3.10"
           - "3.11"
           - "3.12"

.github/workflows/unittests.yml

@@ -14,7 +14,6 @@ jobs:
       fail-fast: false
       matrix:
         python-version:
-          - "3.9"
           - "3.10"
           - "3.11"
           - "3.12"

.pre-commit-config.yaml

@@ -95,7 +95,7 @@ repos:
     rev: v0.13.0
     hooks:
       # Run the linter.
-      - id: ruff
+      - id: ruff-check
         args: [ --fix ]
         # Abort if ruff linter fails as there is some duplication of functionality with
         # the slow pylint hook

.pylintrc

@@ -96,7 +96,7 @@ persistent=yes
 
 # Minimum Python version to use for version dependent checks. Will default to
 # the version used to run pylint.
-py-version=3.9
+py-version=3.10
 
 # Discover python modules and packages in the file system subtree.
 recursive=no

CHANGELOG.md

@@ -22,6 +22,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Removed
 
  - Internal `SilentTQDM` class has been removed; all user-facing functionality has been preserved.
+ - Dropped support for python 3.9 which is near end-of-life.
 
 ### Deprecated
 

pyproject.toml

@@ -7,14 +7,13 @@ name = "coreax"
 dynamic = ["version"]
 description = "Jax coreset algorithms."
 readme = "README.md"
-requires-python = ">=3.9"
+requires-python = ">=3.10"
 keywords = [
     "coreset",
 ]
 classifiers = [
     "Development Status :: 5 - Production/Stable",
     "Programming Language :: Python",
-    "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
@@ -101,7 +100,6 @@ environments = [
     "python_version=='3.12.*'",
     "python_version=='3.11.*'",
     "python_version=='3.10.*'",
-    "python_version=='3.9.*'",
 ]
 
 [project.urls]

tests/unit/test_solvers.py

@@ -28,9 +28,7 @@
     Generic,
     Literal,
     NamedTuple,
-    Optional,
     TypeVar,
-    Union,
     cast,
 )
 from unittest.mock import MagicMock, patch
@@ -91,30 +89,30 @@
 _RefinementSolver = TypeVar("_RefinementSolver", bound=RefinementSolver)
 
 if TYPE_CHECKING:
-    # In Python 3.9-3.10, this raises
+    # In Python 3.10, this raises
     # `TypeError: Multiple inheritance with NamedTuple is not supported`.
     # Thus, we have to do the actual full typing here, and a non-generic one
     # below to be used at runtime.
     class _ReduceProblem(NamedTuple, Generic[_Data, _Solver]):
         dataset: _Data
         solver: _Solver
-        expected_coreset: Optional[AbstractCoreset] = None
+        expected_coreset: AbstractCoreset | None = None
 
     class _RefineProblem(NamedTuple, Generic[_RefinementSolver]):
         initial_coresubset: Coresubset
         solver: _RefinementSolver
-        expected_coresubset: Optional[Coresubset] = None
+        expected_coresubset: Coresubset | None = None
 else:
     # This is the implementation that's used at runtime.
     class _ReduceProblem(NamedTuple):
         dataset: _Data
         solver: _Solver
-        expected_coreset: Optional[AbstractCoreset] = None
+        expected_coreset: AbstractCoreset | None = None
 
     class _RefineProblem(NamedTuple):
         initial_coresubset: Coresubset
         solver: _RefinementSolver
-        expected_coresubset: Optional[Coresubset] = None
+        expected_coresubset: Coresubset | None = None
 
 
 class SolverTest:
@@ -151,7 +149,7 @@ def reduce_problem(
         return _ReduceProblem(Data(dataset), solver, expected_coreset)
 
     def check_solution_invariants(
-        self, coreset: AbstractCoreset, problem: Union[_RefineProblem, _ReduceProblem]
+        self, coreset: AbstractCoreset, problem: _RefineProblem | _ReduceProblem
     ) -> None:
         """
         Check that a coreset obeys certain expected invariant properties.
@@ -796,7 +794,7 @@ def test_functions_impl(x):
 
     @override
     def check_solution_invariants(
-        self, coreset: AbstractCoreset, problem: Union[_RefineProblem, _ReduceProblem]
+        self, coreset: AbstractCoreset, problem: _RefineProblem | _ReduceProblem
     ) -> None:
         r"""
         Check that a coreset obeys certain expected invariant properties.
@@ -1006,7 +1004,7 @@ class ExplicitSizeSolverTest(SolverTest):
 
     @override
     def check_solution_invariants(
-        self, coreset: AbstractCoreset, problem: Union[_RefineProblem, _ReduceProblem]
+        self, coreset: AbstractCoreset, problem: _RefineProblem | _ReduceProblem
     ) -> None:
         super().check_solution_invariants(coreset, problem)
         solver = problem.solver
@@ -1026,7 +1024,7 @@ def check_solution_invariants(
     def test_check_init(
         self,
         solver_factory: jtu.Partial,
-        coreset_size: Union[int, float, str],
+        coreset_size: int | float | str,
         context: AbstractContextManager,
     ) -> None:
         """
@@ -1073,7 +1071,7 @@ def solver_factory(self, request: pytest.FixtureRequest) -> jtu.Partial:
     def reduce_problem(
         self,
         request: pytest.FixtureRequest,
-        solver_factory: Union[type[Solver], jtu.Partial],
+        solver_factory: type[Solver] | jtu.Partial,
     ) -> _ReduceProblem:
         if request.param == "random":
             dataset = jr.uniform(self.random_key, self.shape)
@@ -1707,7 +1705,7 @@ class TestRandomSample(ExplicitSizeSolverTest):
 
     @override
     def check_solution_invariants(
-        self, coreset: AbstractCoreset, problem: Union[_RefineProblem, _ReduceProblem]
+        self, coreset: AbstractCoreset, problem: _RefineProblem | _ReduceProblem
     ) -> None:
         super().check_solution_invariants(coreset, problem)
         solver = cast(RandomSample, problem.solver)
@@ -1730,7 +1728,7 @@ class TestRPCholesky(ExplicitSizeSolverTest):
 
     @override
     def check_solution_invariants(
-        self, coreset: AbstractCoreset, problem: Union[_RefineProblem, _ReduceProblem]
+        self, coreset: AbstractCoreset, problem: _RefineProblem | _ReduceProblem
     ) -> None:
         """Check functionality of 'unique' in addition to the default checks."""
         super().check_solution_invariants(coreset, problem)
@@ -2044,7 +2042,7 @@ def solver_factory(self, request: pytest.FixtureRequest) -> jtu.Partial:
         ],
     )
     def test_regulariser_lambda(
-        self, test_lambda: Optional[Union[float, int]], reduce_problem: _ReduceProblem
+        self, test_lambda: float | int | None, reduce_problem: _ReduceProblem
     ) -> None:
         """Basic checks for the regularisation parameter, lambda."""
         dataset, base_solver, _ = reduce_problem
@@ -2411,7 +2409,7 @@ def solver_factory(self, request) -> jtu.Partial:
     def reduce_problem(
         self,
         request: pytest.FixtureRequest,
-        solver_factory: Union[type[Solver], jtu.Partial],
+        solver_factory: type[Solver] | jtu.Partial,
     ) -> _ReduceProblem:
         if request.param == "random":
             data_key, supervision_key = jr.split(self.random_key)
@@ -2427,7 +2425,7 @@ def reduce_problem(
 
     @override
     def check_solution_invariants(
-        self, coreset: AbstractCoreset, problem: Union[_RefineProblem, _ReduceProblem]
+        self, coreset: AbstractCoreset, problem: _RefineProblem | _ReduceProblem
     ) -> None:
         """Check functionality of 'unique' in addition to the default checks."""
         super().check_solution_invariants(coreset, problem)
@@ -2796,7 +2794,7 @@ def __init__(self, _data: np.ndarray, **kwargs):
                 del kwargs
                 self.data = _data
 
-            def get_arrays(self) -> tuple[Union[np.ndarray, None], ...]:
+            def get_arrays(self) -> tuple[np.ndarray | None, ...]:
                 """Mock sklearn.neighbours.BinaryTree.get_arrays method."""
                 return None, np.arange(len(self.data)), None, None