fix import/wokflow

AMICI-dev · Apr 11, 2024 · a1f37b7 · a1f37b7
1 parent b213adb
commit a1f37b7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/test_benchmark_collection_models.yml b/.github/workflows/test_benchmark_collection_models.yml
@@ -50,7 +50,7 @@ jobs:
       run: |
         pip3 install --user petab[vis] && \
         AMICI_PARALLEL_COMPILE="" pip3 install -v --user \
-            $(ls -t python/sdist/dist/amici-*.tar.gz | head -1)[petab,test,vis]
+            $(ls -t python/sdist/dist/amici-*.tar.gz | head -1)[petab,test,vis,jax]
 
     # retrieve test models
     - name: Download and test benchmark collection

diff --git a/python/sdist/amici/__init__.template.py b/python/sdist/amici/__init__.template.py
@@ -31,7 +31,7 @@ def get_jax_model() -> JAXModel:
 except (ModuleNotFoundError, ImportError):
 
     def get_jax_model() -> JAXModel:
-        raise NotImplementedError()
+        raise NotImplementedError(str(err))
 
 
 __version__ = "TPL_PACKAGE_VERSION"