Fix dim of data for preprocc

havardhhaugen · havardhhaugen · commit 1aacdaffa88d · 2026-05-04T08:55:50.000Z
diff --git a/bris/model/brispredictor.py b/bris/model/brispredictor.py
@@ -160,6 +160,7 @@ def set_static_forcings(
                 dtype=data.dtype,
                 device=data.device,
             )
+            data_input = data_input[None, None, None, ...] # Add bs, time, ens dims needed by imputer
             data_input[..., self.indices[ds]["prognostic_input"]] = data[
                 ..., self.indices[ds]["prognostic_dataset"]
             ]

Original file line number	Diff line number	Diff line change
`@@ -160,6 +160,7 @@ def set_static_forcings(`
`160`	`160`	`dtype=data.dtype,`
`161`	`161`	`device=data.device,`
`162`	`162`	`)`
	`163`	`+ data_input = data_input[None, None, None, ...] # Add bs, time, ens dims needed by imputer`
`163`	`164`	`data_input[..., self.indices[ds]["prognostic_input"]] = data[`
`164`	`165`	`..., self.indices[ds]["prognostic_dataset"]`
`165`	`166`	`]`