..

dweindl · Nov 22, 2023 · 4804cdb · 4804cdb
1 parent 00f6fa6
commit 4804cdb
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/python/sdist/amici/petab_objective.py b/python/sdist/amici/petab_objective.py
@@ -7,6 +7,7 @@
     DeprecationWarning,
 )
 
+from .petab.parameter_mapping import create_parameter_mapping  # noqa: F401
 from .petab.simulations import create_edatas  # noqa: F401
 from .petab.simulations import (  # noqa: F401
     aggregate_sllh,

diff --git a/tests/benchmark-models/test_petab_benchmark.py b/tests/benchmark-models/test_petab_benchmark.py
@@ -8,10 +8,6 @@
 import petab
 import pytest
 from amici.petab.petab_import import import_petab_problem
-from fiddy import MethodId, get_derivative
-from fiddy.derivative_check import NumpyIsCloseDerivativeCheck
-from fiddy.extensions.amici import simulate_petab_to_cached_functions
-from fiddy.success import Consistency
 
 # Absolute and relative tolerances for finite difference gradient checks.
 ATOL: float = 1e-3
@@ -53,12 +49,17 @@
 
 # until fiddy is updated
 @pytest.mark.filterwarnings(
-    "ignore:Importing amici.petab_objective is deprecated.*:DeprecationWarning"
+    "ignore:Importing amici.petab_objective is deprecated.:DeprecationWarning"
 )
 @pytest.mark.filterwarnings("ignore:divide by zero encountered in log10")
 @pytest.mark.parametrize("scale", (True, False))
 @pytest.mark.parametrize("model", models)
 def test_benchmark_gradient(model, scale):
+    from fiddy import MethodId, get_derivative
+    from fiddy.derivative_check import NumpyIsCloseDerivativeCheck
+    from fiddy.extensions.amici import simulate_petab_to_cached_functions
+    from fiddy.success import Consistency
+
     if not scale and model in (
         "Smith_BMCSystBiol2013",
         "Brannmark_JBC2010",