refactor: remove global config edge attribute

ecmwf · Jul 22, 2024 · 62cefe8 · 62cefe8
1 parent 0558563
commit 62cefe8
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/src/anemoi/models/models/encoder_processor_decoder.py b/src/anemoi/models/models/encoder_processor_decoder.py
@@ -52,7 +52,6 @@ def __init__(
         self._graph_data = graph_data
         self._graph_name_data = config.graph.data
         self._graph_name_hidden = config.graph.hidden
-        self.edge_attributes = config.model.attributes.edges
 
         self._calculate_shapes_and_indices(data_indices)
         self._assert_matching_indices(data_indices)
@@ -79,7 +78,6 @@ def __init__(
             in_channels_dst=self.latlons_hidden.shape[1] + self.trainable_hidden_size,
             hidden_dim=self.num_channels,
             sub_graph=self._graph_data[(self._graph_name_data, "to", self._graph_name_hidden)],
-            sub_graph_edge_attributes=self.edge_attributes,
             src_grid_size=self._data_grid_size,
             dst_grid_size=self._hidden_grid_size,
         )
@@ -89,7 +87,6 @@ def __init__(
             config.model.processor,
             num_channels=self.num_channels,
             sub_graph=self._graph_data[(self._graph_name_hidden, "to", self._graph_name_hidden)],
-            sub_graph_edge_attributes=self.edge_attributes,
             src_grid_size=self._hidden_grid_size,
             dst_grid_size=self._hidden_grid_size,
         )
@@ -102,7 +99,6 @@ def __init__(
             hidden_dim=self.num_channels,
             out_channels_dst=self.num_output_channels,
             sub_graph=self._graph_data[(self._graph_name_hidden, "to", self._graph_name_data)],
-            sub_graph_edge_attributes=self.edge_attributes,
             src_grid_size=self._hidden_grid_size,
             dst_grid_size=self._data_grid_size,
         )