Drop Python 3.10

pyproject.toml

@@ -7,7 +7,7 @@ name = 'tjax'
 version = "1.0.9"
 description = 'Tools for JAX.'
 readme = 'README.rst'
-requires-python = '>=3.10,<3.13'
+requires-python = '>=3.11,<3.13'
 license = {file = 'LICENSE'}
 authors = [{email = '[email protected]'}, {name = 'Neil Girdhar'}]
 maintainers = [{email = '[email protected]'}, {name = 'Neil Girdhar'}]
@@ -17,7 +17,6 @@ classifiers = [
   'Operating System :: OS Independent',
   'Programming Language :: Python',
   'Programming Language :: Python :: 3',
-  'Programming Language :: Python :: 3.10',
   'Programming Language :: Python :: 3.11',
   'Programming Language :: Python :: 3.12',
   'Topic :: Scientific/Engineering :: Artificial Intelligence',
@@ -64,7 +63,7 @@ disable = 'ignored-abstractmethod'
 
 [tool.pylint.master]
 jobs = 0
-py-version = '3.10'
+py-version = '3.11'
 load-plugins = [
   'pylint.extensions.bad_builtin',
   'pylint.extensions.broad_try_clause',

tjax/_src/gradient/transform.py

@@ -5,7 +5,8 @@
 
 import jax.numpy as jnp
 from jax import tree
-from typing_extensions import Self, override
+from typing_extensions import override
+from typing import Self
 
 from tjax.dataclasses import dataclass
 

tjax/_src/shims.py

@@ -6,7 +6,8 @@
 
 import jax
 from jax.tree_util import Partial
-from typing_extensions import ParamSpec, Self, override
+from typing_extensions import ParamSpec, override
+from typing import Self
 
 from .function_markers import all_wrapper_assignments