Fix type annotations of task.depends_on and task.produces. (#427)

Author: tobiasraabe

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]

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."""