concise syntax

Signed-off-by: Simon Brugman <[email protected]>
kedro-org · Oct 9, 2023 · 274837f · 274837f
1 parent b146740
commit 274837f
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/kedro-airflow/kedro_airflow/plugin.py b/kedro-airflow/kedro_airflow/plugin.py
@@ -173,11 +173,9 @@ def create(  # noqa: PLR0913
 
         # group memory nodes
         if group_in_memory:
-            catalog = context.catalog
-            nodes, dependencies = group_memory_nodes(catalog, pipeline)
+            nodes, dependencies = group_memory_nodes(context.catalog, pipeline)
         else:
-            nodes = pipeline.nodes
-            nodes = {node.name: [node] for node in nodes}
+            nodes = {node.name: [node] for node in pipeline.nodes}
 
             dependencies = defaultdict(list)
             for node, parent_nodes in pipeline.node_dependencies.items():