build: typing_extensions

Signed-off-by: nstarman <[email protected]>
GalacticDynamics · Jul 17, 2024 · aed1b5b · aed1b5b
1 parent 68528b7
commit aed1b5b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -32,6 +32,7 @@ classifiers = [
 dynamic = ["version"]
 dependencies = [
   "jax",
+  "typing_extensions >= 4.8",
 ]
 
 [project.optional-dependencies]
@@ -140,6 +141,9 @@ ignore = [
   "ISC001",   # Conflicts with formatter
 ]
 
+[tool.ruff.lint.isort]
+extra-standard-library = ["typing_extensions"]
+
 [tool.ruff.lint.per-file-ignores]
 "tests/**" = ["T20"]
 "noxfile.py" = ["T20"]

diff --git a/src/immutable_map_jax/_core.py b/src/immutable_map_jax/_core.py
@@ -8,9 +8,9 @@
 
 from collections.abc import ItemsView, Iterable, Iterator, KeysView, Mapping, ValuesView
 from typing import Annotated, Any, TypeVar, overload
+from typing_extensions import Doc
 
 from jax.tree_util import register_pytree_node_class
-from typing_extensions import Doc
 
 _T = TypeVar("_T")
 K = TypeVar("K")