fix: nodes without attributes

ecmwf · Aug 21, 2024 · db43638 · db43638
1 parent c28e977
commit db43638
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/anemoi/graphs/plotting/interactive_html.py b/src/anemoi/graphs/plotting/interactive_html.py
@@ -179,6 +179,10 @@ def plot_interactive_nodes(graph: HeteroData, nodes_name: str, out_file: Optiona
     # Remove x to avoid plotting the coordinates as an attribute
     node_attrs.remove("x")
 
+    if len(node_attrs) == 0:
+        LOGGER.warning(f"No node attributes found for {nodes_name} nodes.")
+        return
+
     node_traces = {}
     for node_attr in node_attrs:
         node_attr_values = graph[nodes_name][node_attr].float().numpy()