post merge cleanup

AMICI-dev · Oct 19, 2024 · c01f707 · c01f707
1 parent 81e2aeb
commit c01f707
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 104 deletions.
diff --git a/.github/workflows/test_python_cplusplus.yml b/.github/workflows/test_python_cplusplus.yml
@@ -231,11 +231,6 @@ jobs:
     - name: Install python package
       run: scripts/installAmiciSource.sh
 
-    - name: Install notebook dependencies
-      run: |
-        source venv/bin/activate \
-        && pip install jax[cpu]
-
     - name: example notebooks
       run: scripts/runNotebook.sh python/examples/example_*/
 

diff --git a/python/sdist/pyproject.toml b/python/sdist/pyproject.toml
@@ -74,14 +74,21 @@ test = [
     "scipy",
     "pooch"
 ]
-vis =[
+vis = [
     "matplotlib",
     "seaborn",
 ]
-examples =[
+examples = [
     "jupyter",
     "scipy",
 ]
+jax = [
+    "jax>=0.4.34",
+    "jaxlib>=0.4.34",
+    "diffrax>=0.6.0",
+    "equinox>=0.11.8",
+    "optimistix>=0.0.8",
+]
 
 [project.scripts]
 # amici_import_petab.py is kept for backwards compatibility

diff --git a/python/sdist/setup.cfg b/python/sdist/setup.cfg