fix: set default value for optional kwargs in deprecated config

ecmwf · Nov 24, 2024 · 7de0b44 · 7de0b44
1 parent 71b52e7
commit 7de0b44
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/anemoi/graphs/create.py b/src/anemoi/graphs/create.py
@@ -66,8 +66,8 @@ def update_graph(self, graph: HeteroData) -> HeteroData:
 
                 edge_builder_cfg = edges_cfg.get("edge_builder")
                 if edge_builder_cfg is not None:
-                    edge_builder_cfg.source_mask_attr_name = edges_cfg.get("source_mask_attr_name")
-                    edge_builder_cfg.target_mask_attr_name = edges_cfg.get("target_mask_attr_name")
+                    edge_builder_cfg.source_mask_attr_name = edges_cfg.get("source_mask_attr_name", None)
+                    edge_builder_cfg.target_mask_attr_name = edges_cfg.get("target_mask_attr_name", None)
                     edges_cfg.edge_builders = [edge_builder_cfg]
 
             for edge_builder_cfg in edges_cfg.edge_builders: