actually generate code

AMICI-dev · Apr 10, 2024 · 37cdc81 · 37cdc81
1 parent 5974d47
commit 37cdc81
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/sdist/amici/de_export.py b/python/sdist/amici/de_export.py
@@ -246,6 +246,7 @@ def generate_model_code(self) -> None:
             sp.Pow, "_eval_derivative", _custom_pow_eval_derivative
         ):
             self._prepare_model_folder()
+            self._generate_jax_code()
             self._generate_c_code()
             self._generate_m_code()