From 6c03866ee39f345585d4404e3763f679d9d49ee0 Mon Sep 17 00:00:00 2001 From: Cor Zuurmond Date: Tue, 29 Oct 2024 14:10:34 +0100 Subject: [PATCH] Fix create node queue --- src/databricks/labs/ucx/assessment/sequencing.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/databricks/labs/ucx/assessment/sequencing.py b/src/databricks/labs/ucx/assessment/sequencing.py index 321b01ddce..d54c96be41 100644 --- a/src/databricks/labs/ucx/assessment/sequencing.py +++ b/src/databricks/labs/ucx/assessment/sequencing.py @@ -263,6 +263,6 @@ def _create_node_queue(self, incoming: dict[MigrationNodeKey, set[MigrationNode] first. """ priority_queue = PriorityQueue() - for node_key, incoming_nodes in incoming.items(): - priority_queue.put(len(incoming_nodes), self._nodes[node_key]) + for node_key, node in self._nodes.items(): + priority_queue.put(len(incoming[node_key]), self._nodes[node_key]) return priority_queue