Fix MLP initialization (ecmwf#2052)

kacpnowak · web-flow · commit 28c76eef2617 · 2026-03-13T13:03:01.000+01:00
diff --git a/src/weathergen/model/model.py b/src/weathergen/model/model.py
@@ -437,7 +437,7 @@ def create(self) -> "Model":
                             with_residual=False,
                             dropout_rate=dropout_rate,
                             norm_eps=self.cf.mlp_norm_eps,
-                            stream_name=f"embed_target_coords_{stream_name}",
+                            name=f"embed_target_coords_{stream_name}",
                         )
                     else:
                         assert False

Original file line number	Diff line number	Diff line change
`@@ -437,7 +437,7 @@ def create(self) -> "Model":`
`437`	`437`	`with_residual=False,`
`438`	`438`	`dropout_rate=dropout_rate,`
`439`	`439`	`norm_eps=self.cf.mlp_norm_eps,`
`440`		`- stream_name=f"embed_target_coords_{stream_name}",`
	`440`	`+ name=f"embed_target_coords_{stream_name}",`
`441`	`441`	`)`
`442`	`442`	`else:`
`443`	`443`	`assert False`