Add typing_extensions to dependencies

brentyi · May 5, 2024 · b316405 · b316405
1 parent f172a88
commit b316405
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -20,6 +20,7 @@
         "jax>=0.3.18",  # For jax.Array.
         "jax_dataclasses>=1.4.4",
         "numpy",
+        "typing_extensions>=4.0.0",
         "tyro",  # Only used in examples.
     ],
     extras_require={