fix

src/anemoi/models/models/encoder_processor_decoder.py

@@ -48,8 +48,6 @@ def __init__(
             Graph definition
         """
         super().__init__()
-        print('❌', graph_data)
-        print('❌', data_indices)
 
         self._graph_data = graph_data
         self._graph_name_data = config.graph.data
@@ -142,7 +140,7 @@ def _register_latlon(self, name: str, nodes: str) -> None:
             Name of nodes to map
         """
         print(nodes)
-        print('❌', self._graph_data[nodes])
+        # print(self._graph_data[nodes])
         coords = self._graph_data[nodes].x
         sin_cos_coords = torch.cat([torch.sin(coords), torch.cos(coords)], dim=-1)
         self.register_buffer(f"latlons_{name}", sin_cos_coords, persistent=True)
@@ -278,11 +276,4 @@ def forward(self, x: Tensor, model_comm_group: Optional[ProcessGroup] = None) ->
 
         # residual connection (just for the prognostic variables)
         x_out[..., self._internal_output_idx] += x[:, -1, :, :, self._internal_input_idx]
-        return x_out
-
-# class AnemoiModelEncProcDec(nn.Module):
-#     def __init__(self, *, config, data_indices, graph_data):
-#         super().__init__()
-#     def forward(self, x, model_comm_group=None):
-#         print('❌', type(x))
-#         return x
+        return x_out