Merge branch 'main' into kedro-dvc-documentation

kedro-org · Feb 6, 2025 · d957b8d · d957b8d
2 parents 80822de + a064c46
commit d957b8d
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/kedro/runner/runner.py b/kedro/runner/runner.py
@@ -11,7 +11,13 @@
 import warnings
 from abc import ABC, abstractmethod
 from collections import Counter, deque
-from concurrent.futures import FIRST_COMPLETED, Executor, ProcessPoolExecutor, wait
+from concurrent.futures import (
+    FIRST_COMPLETED,
+    Executor,
+    Future,
+    ProcessPoolExecutor,
+    wait,
+)
 from itertools import chain
 from typing import TYPE_CHECKING, Any
 
@@ -258,7 +264,7 @@ def _raise_runtime_error(
         todo_nodes: set[Node],
         done_nodes: set[Node],
         ready: set[Node],
-        done: set[Node] | None,
+        done: set[Future[Node]] | None,
     ) -> None:
         debug_data = {
             "todo_nodes": todo_nodes,