Adapt to recent amici/petab changes (#47)

ICB-DCM · Oct 8, 2024 · 6aea82e · 6aea82e
1 parent d95a40d
commit 6aea82e
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/extensions/amici/test_amici.py b/tests/extensions/amici/test_amici.py
@@ -3,10 +3,10 @@
 from pathlib import Path
 
 import amici
-import amici.petab_import
-import amici.petab_objective
+import amici.petab.petab_import
+import amici.petab.simulations
 import numpy as np
-import petab
+import petab.v1 as petab
 import pytest
 
 from fiddy import MethodId, Type, get_derivative
@@ -54,7 +54,7 @@ def simple() -> petab.Problem:
 def test_run_amici_simulation_to_functions(problem_generator):
     petab_problem, point = problem_generator()
     timepoints = sorted(set(petab_problem.measurement_df.time))
-    amici_model = amici.petab_import.import_petab_problem(petab_problem)
+    amici_model = amici.petab.petab_import.import_petab_problem(petab_problem)
     amici_model.setTimepoints(timepoints)
     amici_solver = amici_model.getSolver()
 
@@ -120,7 +120,7 @@ def test_run_amici_simulation_to_functions(problem_generator):
 @pytest.mark.parametrize("scaled_parameters", (False, True))
 def test_simulate_petab_to_functions(problem_generator, scaled_parameters):
     petab_problem, point = problem_generator()
-    amici_model = amici.petab_import.import_petab_problem(petab_problem)
+    amici_model = amici.petab.petab_import.import_petab_problem(petab_problem)
     amici_solver = amici_model.getSolver()
 
     if amici_model.getName() == "simple":