Update autoemulate/core/metrics.py

radka-j · sgreenbury · web-flow · commit cdc8111920ab · 2025-11-18T11:08:07.000Z
Co-authored-by: Sam Greenbury &lt;50113363+sgreenbury@users.noreply.github.com&gt;
diff --git a/autoemulate/core/metrics.py b/autoemulate/core/metrics.py
@@ -373,7 +373,7 @@ def __call__(
                         "distributions."
                     )
                     raise ValueError(msg)
-                return model_nll_output
+                return model_nll_output.reshape(*y_true.shape)
             msg = (
                 f"Unknown reduction method: {metric_params.reduction}. "
                 "Expected 'mean' or 'none'."

Original file line number	Diff line number	Diff line change
`@@ -373,7 +373,7 @@ def __call__(`
`373`	`373`	`"distributions."`
`374`	`374`	`)`
`375`	`375`	`raise ValueError(msg)`
`376`		`- return model_nll_output`
	`376`	`+ return model_nll_output.reshape(*y_true.shape)`
`377`	`377`	`msg = (`
`378`	`378`	`f"Unknown reduction method: {metric_params.reduction}. "`
`379`	`379`	`"Expected 'mean' or 'none'."`