Skip to content

Commit

Permalink
feat: add ndim to attribute name
Browse files Browse the repository at this point in the history
  • Loading branch information
JPXKQX committed Aug 21, 2024
1 parent db43638 commit 67546a4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/anemoi/graphs/descriptor.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 67546a4

Please sign in to comment.