Update src/anemoi/graphs/edges/attributes.py

Co-authored-by: Mario Santa Cruz <48736305+JPXKQX@users.noreply.github.com>
MeteoSwiss · Dec 18, 2024 · 40394e4 · 40394e4
1 parent 919d0ad
commit 40394e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/anemoi/graphs/edges/attributes.py b/src/anemoi/graphs/edges/attributes.py
@@ -202,7 +202,7 @@ def get_raw_values(self, graph: HeteroData, source_name: str, target_name: str)
 
         edge_index = graph[(source_name, "to", target_name)].edge_index
         assert hasattr(graph[node_name], self.node_attr_name)
-        val = getattr(graph[node_name], self.node_attr_name).numpy()[edge_index[self.idx]]
+        return graph[node_name][self.node_attr_name].numpy()[edge_index[self.idx]]
         return val