This commit is contained in:
Nuno Campos
2025-05-08 16:49:38 -07:00
parent 6c155f87c3
commit d04570f178
+4
View File
@@ -387,6 +387,8 @@ def prepare_next_tasks(
manager: Literal[None] = None,
trigger_to_nodes: Optional[Mapping[str, Sequence[str]]] = None,
updated_channels: Optional[set[str]] = None,
retry_policy: Sequence[RetryPolicy] = (),
cache_policy: Literal[None] = None,
) -> dict[str, PregelTask]: ...
@@ -406,6 +408,8 @@ def prepare_next_tasks(
manager: Union[None, ParentRunManager, AsyncParentRunManager],
trigger_to_nodes: Optional[Mapping[str, Sequence[str]]] = None,
updated_channels: Optional[set[str]] = None,
retry_policy: Sequence[RetryPolicy] = (),
cache_policy: Optional[CachePolicy] = None,
) -> dict[str, PregelExecutableTask]: ...