Fix create node queue

databrickslabs · Oct 29, 2024 · 6c03866 · 6c03866
1 parent 477292a
commit 6c03866
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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