diff --git a/src/_pytask/node_protocols.py b/src/_pytask/node_protocols.py index 8eee22c9..78c9deb3 100644 --- a/src/_pytask/node_protocols.py +++ b/src/_pytask/node_protocols.py @@ -64,8 +64,8 @@ class PTask(MetaNode, Protocol): """Protocol for nodes.""" name: str - depends_on: PyTree[PNode] - produces: PyTree[PNode] + depends_on: dict[str, PyTree[PNode]] + produces: dict[str, PyTree[PNode]] markers: list[Mark] report_sections: list[tuple[str, str, str]] attributes: dict[Any, Any] diff --git a/src/_pytask/nodes.py b/src/_pytask/nodes.py index 8862696d..cc0dc280 100644 --- a/src/_pytask/nodes.py +++ b/src/_pytask/nodes.py @@ -39,9 +39,9 @@ class TaskWithoutPath(PTask): """The base name of the task.""" function: Callable[..., Any] """The task function.""" - depends_on: PyTree[PNode] = field(factory=dict) + depends_on: dict[str, PyTree[PNode]] = field(factory=dict) """A list of dependencies of task.""" - produces: PyTree[PNode] = field(factory=dict) + produces: dict[str, PyTree[PNode]] = field(factory=dict) """A list of products of task.""" markers: list[Mark] = field(factory=list) """A list of markers attached to the task function.""" @@ -78,9 +78,9 @@ class Task(PTaskWithPath): """The name of the task.""" display_name: str | None = field(default=None, init=False) """The shortest uniquely identifiable name for task for display.""" - depends_on: PyTree[PNode] = field(factory=dict) + depends_on: dict[str, PyTree[PNode]] = field(factory=dict) """A list of dependencies of task.""" - produces: PyTree[PNode] = field(factory=dict) + produces: dict[str, PyTree[PNode]] = field(factory=dict) """A list of products of task.""" markers: list[Mark] = field(factory=list) """A list of markers attached to the task function."""