Skip to content

Commit c3aa401

Browse files
committed
forgot to add a baseline-renom change to opt
1 parent 48b6b96 commit c3aa401

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

hnn_core/optimization/objective_functions.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,17 @@ def _rmse_evoked(
5252
new_net = initial_net.copy()
5353
set_params(new_net, params)
5454

55-
dpls = simulate_dipole(new_net, tstop=tstop, n_trials=obj_fun_kwargs["n_trials"])
55+
if "bsl_cor" in obj_fun_kwargs:
56+
dpls = simulate_dipole(
57+
new_net,
58+
tstop=tstop,
59+
n_trials=obj_fun_kwargs["n_trials"],
60+
bsl_cor=obj_fun_kwargs["bsl_cor"],
61+
)
62+
else:
63+
dpls = simulate_dipole(
64+
new_net, tstop=tstop, n_trials=obj_fun_kwargs["n_trials"]
65+
)
5666

5767
# smooth & scale
5868
if "scale_factor" in obj_fun_kwargs:

0 commit comments

Comments
 (0)