Fix callback usage in TraceFitter.refine

brian-team · Nov 16, 2020 · 88093d6 · 88093d6
1 parent c301da9
commit 88093d6
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/brian2modelfitting/fitter.py b/brian2modelfitting/fitter.py
@@ -1279,10 +1279,9 @@ def _callback_wrapper(params, iter, resid, *args, **kwds):
                 params = {p: Quantity(val, dim=self.model[p].dim)
                           for p, val in params.items()}
                 best_raw_error = tuple([Quantity(raw_error,
-                                                 dim=metric.get_dimensions(output_dim))
-                                        for raw_error, metric, output_dim
+                                                 dim=output_dim**2)
+                                        for raw_error, output_dim
                                         in zip(errors[best_idx],
-                                               self.metric,
                                                self.output_dim)])
             else:
                 all_errors = array(combined_errors)