From 62cefe86b2cab0de244748fcc980e10ffd5c7501 Mon Sep 17 00:00:00 2001 From: Jesper Dramsch Date: Mon, 22 Jul 2024 12:51:23 +0000 Subject: [PATCH] refactor: remove global config edge attribute --- src/anemoi/models/models/encoder_processor_decoder.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/src/anemoi/models/models/encoder_processor_decoder.py b/src/anemoi/models/models/encoder_processor_decoder.py index 6eb81c8..0f37474 100644 --- 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, )