fix build

mara · Feb 24, 2023 · 9dea418 · 9dea418
1 parent 6fa24ba
commit 9dea418
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mara_pipelines/logging/node_cost.py b/mara_pipelines/logging/node_cost.py
@@ -9,7 +9,7 @@
 from .. import pipelines
 
 
-def node_durations_and_run_times(node: pipelines.Node) -> Dict[tuple, [float, float]]:
+def node_durations_and_run_times(node: pipelines.Node) -> Dict[tuple, object]:  # object = [float, float]
     """
     Returns for children of `node` the average duration and run time (sum of average duration of all leaf nodes)
 
@@ -44,7 +44,7 @@ def node_durations_and_run_times(node: pipelines.Node) -> Dict[tuple, [float, fl
         return {tuple(row[0]): row[1:] for row in cursor.fetchall()}
 
 
-def compute_cost(node: pipelines.Node, node_durations_and_run_times: Dict[tuple, [float, float]]) -> float:
+def compute_cost(node: pipelines.Node, node_durations_and_run_times: Dict[tuple, object]) -> float:  # object = [float, float]
     """
     Computes the cost of a node as maximum cumulative run time of a node and all its downstreams.
     Stores the result in `node` and also returns it