add jaxtyping dep.

meom-group · Feb 9, 2024 · 75aad62 · 75aad62
1 parent 38d5470
commit 75aad62
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -28,6 +28,7 @@ dependencies = [
     "xarray",
     "jax",
     "jaxlib",
+    "jaxtyping",
     "optax"
 ]
 dynamic = ["version"]