diff --git a/pyiron_workflow/nodes/macro.py b/pyiron_workflow/nodes/macro.py index ced1e533..fde7ab66 100644 --- a/pyiron_workflow/nodes/macro.py +++ b/pyiron_workflow/nodes/macro.py @@ -273,7 +273,9 @@ def _setup_node(self) -> None: @staticmethod @abstractmethod - def graph_creator(self, *args, **kwargs) -> Callable: + def graph_creator( + self: Macro, *args, **kwargs + ) -> HasChannel | tuple[HasChannel, ...] | None: """Build the graph the node will run.""" @classmethod @@ -538,7 +540,7 @@ def decorator(graph_creator): def macro_node( - graph_creator: Callable, + graph_creator: Callable[..., HasChannel | tuple[HasChannel, ...] | None], *node_args, output_labels: str | tuple[str, ...] | None = None, validate_output_labels: bool = True,