refactor: remove second return

ecmwf · Sep 2, 2024 · 5d247de · 5d247de
1 parent 19e38b0
commit 5d247de
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/anemoi/graphs/edges/builder.py b/src/anemoi/graphs/edges/builder.py
@@ -119,10 +119,8 @@ def update_graph(self, graph: HeteroData, attrs_config: DotDict | None = None) -
         """
         graph = self.register_edges(graph)
 
-        if attrs_config is None:
-            return graph
-
-        graph = self.register_attributes(graph, attrs_config)
+        if attrs_config is not None:
+            graph = self.register_attributes(graph, attrs_config)
 
         return graph