Skip to content

Commit

Permalink
use AbstractVertex rather than Union
Browse files Browse the repository at this point in the history
  • Loading branch information
Christian-B committed Jan 7, 2025
1 parent f125ce4 commit aa39b4e
Showing 1 changed file with 1 addition and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -591,8 +591,7 @@ def get_in_coming_vertices(self, partition_id: str) -> Sequence[
def get_source_specific_in_coming_vertices(
self, source_vertex: ApplicationVertex,
partition_id: str) -> Sequence[
Tuple[MachineVertex, Sequence[Union[
MachineVertex, ApplicationVertex]]]]:
Tuple[MachineVertex, Sequence[AbstractVertex]]]:
# If delayed get the real pre-vertex
if isinstance(source_vertex, DelayExtensionVertex):
pre_vertex = cast(
Expand Down

0 comments on commit aa39b4e

Please sign in to comment.