Ruff

havardhhaugen · havardhhaugen · commit ebeed32b0ecb · 2026-05-04T08:56:57.000Z
diff --git a/bris/model/brispredictor.py b/bris/model/brispredictor.py
@@ -160,7 +160,9 @@ 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 = 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,7 +160,9 @@ 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`	`+ data_input = data_input[`
	`164`	`+ None, None, None, ...`
	`165`	`+ ] # Add bs, time, ens dims needed by imputer`
`164`	`166`	`data_input[..., self.indices[ds]["prognostic_input"]] = data[`
`165`	`167`	`..., self.indices[ds]["prognostic_dataset"]`
`166`	`168`	`]`