Apply writes / subgraph states to tasks to be returned in a StateSnapshot.
tasks_w_writes(
tasks: Iterable[PregelTask | PregelExecutableTask],
pending_writes: list[PendingWrite] | None,
states: dict[str, RunnableConfig | StateSnapshot] | None,
output_keys: str | Sequence[str]
) -> tuple[PregelTask, ...]