Adjust docs.

Author: tobiasraabe

docs_src/how_to_guides/writing_custom_nodes_example_3_py310.py

@@ -15,12 +15,20 @@ class PickleNode:
         Name of the node which makes it identifiable in the DAG.
     path
         The path to the file.
+    attributes
+        Additional attributes that are stored in the node.
 
     """
 
-    def __init__(self, name: str = "", path: Path | None = None) -> None:
+    def __init__(
+        self,
+        name: str = "",
+        path: Path | None = None,
+        attributes: dict[Any, Any] | None = None,
+    ) -> None:
         self.name = name
         self.path = path
+        self.attributes = attributes or {}
 
     @property
     def signature(self) -> str:

docs_src/how_to_guides/writing_custom_nodes_example_3_py38.py

@@ -16,12 +16,20 @@ class PickleNode:
         Name of the node which makes it identifiable in the DAG.
     path
         The path to the file.
+    attributes
+        Additional attributes that are stored in the node.
 
     """
 
-    def __init__(self, name: str = "", path: Optional[Path] = None) -> None:
+    def __init__(
+        self,
+        name: str = "",
+        path: Optional[Path] = None,
+        attributes: Optional[dict[Any, Any]] = None,
+    ) -> None:
         self.name = name
         self.path = path
+        self.attributes = attributes or {}
 
     @property
     def signature(self) -> str:

tests/test_collect_command.py

@@ -517,6 +517,7 @@ def task_example(
 def test_node_protocol_for_custom_nodes_with_paths(runner, tmp_path):
     source = """
     from typing import Annotated
+    from typing import Any
     from pytask import Product
     from pathlib import Path
     from attrs import define
@@ -527,6 +528,7 @@ class PickleFile:
         name: str
         path: Path
         signature: str = "id"
+        attributes: dict[Any, Any] = {}
 
         def state(self):
             return str(self.path.stat().st_mtime)