Skip to content

Commit f86d56b

Browse files
authored
Fix #2969 (#2973)
Missed a few lines in #2969.
1 parent ef7f718 commit f86d56b

File tree

2 files changed

+46
-46
lines changed

2 files changed

+46
-46
lines changed

tests/performance/test.py

Lines changed: 43 additions & 43 deletions
Original file line numberDiff line numberDiff line change
@@ -29,26 +29,26 @@ def parse_args():
2929
def check_results(rdata):
3030
diagnostics = [
3131
"numsteps",
32-
"numstepsB",
33-
"numrhsevals",
34-
"numrhsevalsB",
35-
"numerrtestfails",
36-
"numerrtestfailsB",
37-
"numnonlinsolvconvfails",
38-
"numnonlinsolvconvfailsB",
32+
"numsteps_b",
33+
"num_rhs_evals",
34+
"num_rhs_evals_b",
35+
"num_err_test_fails",
36+
"num_err_test_fails_b",
37+
"num_non_lin_solv_conv_fails",
38+
"num_non_lin_solv_conv_fails_b",
3939
"preeq_status",
4040
"preeq_numsteps",
41-
"preeq_numstepsB",
41+
"preeq_numsteps_b",
4242
"preeq_cpu_time",
43-
"preeq_cpu_timeB",
43+
"preeq_cpu_time_b",
4444
"cpu_time_total",
4545
"cpu_time",
46-
"cpu_timeB",
46+
"cpu_time_b",
4747
"posteq_status",
4848
"posteq_cpu_time",
49-
"posteq_cpu_timeB",
49+
"posteq_cpu_time_b",
5050
"posteq_numsteps",
51-
"posteq_numstepsB",
51+
"posteq_numsteps_b",
5252
]
5353
for d in diagnostics:
5454
print(d, rdata[d])
@@ -94,40 +94,40 @@ def compile_model(model_dir_source: Path, model_dir_compiled: Path):
9494

9595
def prepare_simulation(arg, model, solver, edata):
9696
if arg == "forward_simulation":
97-
solver.setSensitivityMethod(amici.SensitivityMethod.none)
98-
solver.setSensitivityOrder(amici.SensitivityOrder.none)
97+
solver.set_sensitivity_method(amici.SensitivityMethod.none)
98+
solver.set_sensitivity_order(amici.SensitivityOrder.none)
9999
elif arg == "forward_sensitivities":
100-
model.setParameterList(list(range(100)))
101-
solver.setSensitivityMethod(amici.SensitivityMethod.forward)
102-
solver.setSensitivityOrder(amici.SensitivityOrder.first)
100+
model.set_parameter_list(list(range(100)))
101+
solver.set_sensitivity_method(amici.SensitivityMethod.forward)
102+
solver.set_sensitivity_order(amici.SensitivityOrder.first)
103103
elif arg == "adjoint_sensitivities":
104-
solver.setSensitivityMethod(amici.SensitivityMethod.adjoint)
105-
solver.setSensitivityOrder(amici.SensitivityOrder.first)
104+
solver.set_sensitivity_method(amici.SensitivityMethod.adjoint)
105+
solver.set_sensitivity_order(amici.SensitivityOrder.first)
106106
elif arg == "forward_simulation_non_optimal_parameters":
107-
tmp_par = model.getParameters()
108-
model.setParameters([0.1 for _ in tmp_par])
109-
solver.setSensitivityMethod(amici.SensitivityMethod.none)
110-
solver.setSensitivityOrder(amici.SensitivityOrder.none)
107+
tmp_par = model.get_parameters()
108+
model.set_parameters([0.1 for _ in tmp_par])
109+
solver.set_sensitivity_method(amici.SensitivityMethod.none)
110+
solver.set_sensitivity_order(amici.SensitivityOrder.none)
111111
elif arg == "adjoint_sensitivities_non_optimal_parameters":
112-
tmp_par = model.getParameters()
113-
model.setParameters([0.1 for _ in tmp_par])
114-
solver.setSensitivityMethod(amici.SensitivityMethod.adjoint)
115-
solver.setSensitivityOrder(amici.SensitivityOrder.first)
112+
tmp_par = model.get_parameters()
113+
model.set_parameters([0.1 for _ in tmp_par])
114+
solver.set_sensitivity_method(amici.SensitivityMethod.adjoint)
115+
solver.set_sensitivity_order(amici.SensitivityOrder.first)
116116
elif arg == "forward_steadystate_sensitivities_non_optimal_parameters":
117-
tmp_par = model.getParameters()
118-
model.setParameters([0.1 for _ in tmp_par])
119-
solver.setSensitivityMethod(amici.SensitivityMethod.forward)
120-
solver.setSensitivityOrder(amici.SensitivityOrder.first)
121-
model.setSteadyStateSensitivityMode(
117+
tmp_par = model.get_parameters()
118+
model.set_parameters([0.1 for _ in tmp_par])
119+
solver.set_sensitivity_method(amici.SensitivityMethod.forward)
120+
solver.set_sensitivity_order(amici.SensitivityOrder.first)
121+
model.set_steady_state_sensitivity_mode(
122122
amici.SteadyStateSensitivityMode.newtonOnly
123123
)
124124
edata.setTimepoints([float("inf")])
125125
elif arg == "adjoint_steadystate_sensitivities_non_optimal_parameters":
126-
tmp_par = model.getParameters()
127-
model.setParameters([0.1 for _ in tmp_par])
128-
solver.setSensitivityMethod(amici.SensitivityMethod.adjoint)
129-
solver.setSensitivityOrder(amici.SensitivityOrder.first)
130-
edata.setTimepoints([float("inf")])
126+
tmp_par = model.get_parameters()
127+
model.set_parameters([0.1 for _ in tmp_par])
128+
solver.set_sensitivity_method(amici.SensitivityMethod.adjoint)
129+
solver.set_sensitivity_order(amici.SensitivityOrder.first)
130+
edata.set_timepoints([float("inf")])
131131
else:
132132
print("Unknown argument:", arg)
133133
sys.exit(1)
@@ -153,17 +153,17 @@ def main():
153153
model_module = amici.import_model_module(
154154
model_name, model_dir_compiled
155155
)
156-
model = model_module.getModel()
157-
solver = model.getSolver()
156+
model = model_module.get_model()
157+
solver = model.create_solver()
158158
# TODO
159159
edata = amici.ExpData(model)
160-
edata.setTimepoints([1e8])
161-
edata.setObservedData([1.0])
162-
edata.setObservedDataStdDev([1.0])
160+
edata.set_timepoints([1e8])
161+
edata.set_observed_data([1.0])
162+
edata.set_observed_data_std_dev([1.0])
163163

164164
prepare_simulation(arg, model, solver, edata)
165165

166-
rdata = amici.runAmiciSimulation(model, solver, edata)
166+
rdata = amici.run_simulation(model, solver, edata)
167167

168168
check_results(rdata)
169169

tests/sbml/testSBMLSuiteJax.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,17 +34,17 @@ def __init__(self, jax_model: object, importer: amici.SbmlImporter):
3434
self.jax_model = jax_model
3535
self.importer = importer
3636

37-
def getParameterIds(self):
37+
def get_parameter_ids(self):
3838
return list(self.jax_model.parameter_ids)
3939

40-
def getParameterById(self, pid: str):
40+
def get_parameter_by_id(self, pid: str):
4141
return float(
4242
self.jax_model.parameters[
4343
list(self.jax_model.parameter_ids).index(pid)
4444
]
4545
)
4646

47-
def getExpressionIds(self):
47+
def get_expression_ids(self):
4848
return list(self.jax_model.expression_ids)
4949

5050

0 commit comments

Comments
 (0)