[fix] update dependencies

paninski-lab · Jun 27, 2024 · bab5bc9 · bab5bc9
1 parent 493a75b
commit bab5bc9
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -25,17 +25,18 @@ def get_version(rel_path):
 # basic requirements
 install_requires = [
     'ipykernel',
+    'jax',
+    'jaxlib',
     'matplotlib',
     'numpy',
     'opencv-python',
+    'optax',
     'pandas',
     'scikit-learn',
     'scipy>=1.2.0',
+    'sleap_io',
     'tqdm',
     'typing',
-    'sleap_io',
-    'jax',
-    'jaxlib',
 ]
 
 # additional requirements