fix tests

chaobrain · Nov 7, 2024 · 1976abd · 1976abd
1 parent 2e47a28
commit 1976abd
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -48,6 +48,7 @@ dependencies = [
     'jax',
     'jaxlib',
     'numpy',
+    'typing_extensions',
 ]
 
 dynamic = ['version']

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,5 @@
 numpy
 jax
 jaxlib
+typing_extensions
 
diff --git a/setup.py b/setup.py
@@ -56,7 +56,7 @@
   author_email='chao.brain@qq.com',
   packages=packages,
   python_requires='>=3.9',
-  install_requires=['numpy>=1.15', 'jax'],
+  install_requires=['numpy>=1.15', 'jax', 'typing_extensions'],
   url='https://github.com/chaoming0625/brainunit',
   project_urls={
     "Bug Tracker": "https://github.com/chaoming0625/brainunit/issues",