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
 

benchmark/blobs_benchmark.py

@@ -31,7 +31,6 @@
 import json
 import os
 import time
-from typing import Union
 
 import jax
 import jax.numpy as jnp
@@ -96,8 +95,8 @@ def setup_stein_kernel(
 
     # Define the score function as the gradient of log density given by the KDE
     def score_function(
-        x: Union[Shaped[Array, " n d"], Shaped[Array, ""], float, int],
-    ) -> Union[Shaped[Array, " n d"], Shaped[Array, " 1 1"]]:
+        x: Shaped[Array, " n d"] | Shaped[Array, ""] | float | int,
+    ) -> Shaped[Array, " n d"] | Shaped[Array, " 1 1"]:
         """
         Compute the score function (gradient of log density) for a single point.
 

benchmark/david_benchmark.py

@@ -38,7 +38,6 @@
 import math
 import time
 from pathlib import Path
-from typing import Optional
 
 import equinox as eqx
 import jax.numpy as jnp
@@ -60,7 +59,7 @@
 # (disable line too long and too many statements ruff)
 def benchmark_coreset_algorithms(
     in_path: Path = Path("../examples/data/david_orig.png"),
-    out_path: Optional[Path] = Path(
+    out_path: Path | None = Path(
         "../examples/benchmarking_images/david_benchmark_results.png"
     ),
     downsampling_factor: int = 1,

benchmark/mnist_benchmark.py

@@ -40,7 +40,7 @@
 import json
 import os
 import time
-from typing import Any, NamedTuple, Optional, Union
+from typing import Any, NamedTuple
 
 import equinox as eqx
 import jax
@@ -143,7 +143,7 @@ def __call__(self, x: jnp.ndarray, training: bool = True) -> jnp.ndarray:
 class TrainState(train_state.TrainState):
     """Custom train state with batch statistics and dropout RNG."""
 
-    batch_stats: Optional[dict[str, jnp.ndarray]]
+    batch_stats: dict[str, jnp.ndarray] | None
     dropout_rng: KeyArrayLike
 
 
@@ -416,7 +416,7 @@ def prepare_datasets() -> tuple[jnp.ndarray, jnp.ndarray, jnp.ndarray, jnp.ndarr
 def train_model(
     data_bundle: dict[str, jnp.ndarray],
     key: KeyArrayLike,
-    config: dict[str, Union[int, float]],
+    config: dict[str, int | float],
 ) -> dict[str, float]:
     """
     Train the model and return the results.

coreax/approximation.py

@@ -30,14 +30,14 @@
 
 from collections.abc import Callable
 from functools import partial
-from typing import TYPE_CHECKING, Any, Union
+from typing import TYPE_CHECKING, Any, Literal
 
 import jax
 import jax.numpy as jnp
 import jax.random as jr
 from jax import Array
 from jaxtyping import Shaped
-from typing_extensions import Literal, override
+from typing_extensions import override
 
 from coreax.data import Data, _atleast_2d_consistent
 from coreax.kernels import UniCompositeKernel
@@ -177,14 +177,12 @@ class MonteCarloApproximateKernel(RandomRegressionKernel):
 
     def gramian_row_mean(
         self,
-        x: Union[
-            Shaped[Array, " n d"],
-            Shaped[Array, " d"],
-            Shaped[Array, ""],
-            float,
-            int,
-            Data,
-        ],
+        x: Shaped[Array, " n d"]
+        | Shaped[Array, " d"]
+        | Shaped[Array, ""]
+        | float
+        | int
+        | Data,
         **kwargs: Any,
     ) -> Shaped[Array, " n"]:
         r"""
@@ -232,14 +230,12 @@ class ANNchorApproximateKernel(RandomRegressionKernel):
 
     def gramian_row_mean(
         self,
-        x: Union[
-            Shaped[Array, " n d"],
-            Shaped[Array, " d"],
-            Shaped[Array, ""],
-            float,
-            int,
-            Data,
-        ],
+        x: Shaped[Array, " n d"]
+        | Shaped[Array, " d"]
+        | Shaped[Array, ""]
+        | float
+        | int
+        | Data,
         **kwargs: Any,
     ) -> Shaped[Array, " n"]:
         r"""
@@ -305,14 +301,12 @@ class NystromApproximateKernel(RandomRegressionKernel):
 
     def gramian_row_mean(
         self,
-        x: Union[
-            Shaped[Array, " n d"],
-            Shaped[Array, " d"],
-            Shaped[Array, ""],
-            float,
-            int,
-            Data,
-        ],
+        x: Shaped[Array, " n d"]
+        | Shaped[Array, " d"]
+        | Shaped[Array, ""]
+        | float
+        | int
+        | Data,
         **kwargs: Any,
     ) -> Shaped[Array, " n"]:
         r"""