diff --git a/src/anemoi/graphs/descriptor.py b/src/anemoi/graphs/descriptor.py index 2232fbb..dde565c 100644 --- a/src/anemoi/graphs/descriptor.py +++ b/src/anemoi/graphs/descriptor.py @@ -91,7 +91,7 @@ def get_edge_summary(self) -> list[list]: self.graph[src_name].num_nodes - len(torch.unique(edges.edge_index[0])), self.graph[dst_name].num_nodes - len(torch.unique(edges.edge_index[1])), sum(edges[attr].shape[1] for attr in attributes), - ", ".join(attributes), + ", ".join([f"{attr}({edges[attr].shape[1]}D)" for attr in attributes]), ] ) return edge_summary