Skip to content

Commit

Permalink
formatting fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
farridav committed Jan 2, 2025
1 parent a72f17f commit 663dd90
Showing 1 changed file with 5 additions and 8 deletions.
13 changes: 5 additions & 8 deletions sdk/python/kfp/compiler/compiler_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,12 +19,7 @@
from typing import DefaultDict, Dict, List, Mapping, Set, Tuple, Union

from kfp import dsl
from kfp.dsl import constants
from kfp.dsl import for_loop
from kfp.dsl import pipeline_channel
from kfp.dsl import pipeline_context
from kfp.dsl import pipeline_task
from kfp.dsl import tasks_group
from kfp.dsl import constants, for_loop, pipeline_channel, pipeline_context, pipeline_task, tasks_group

GroupOrTaskType = Union[tasks_group.TasksGroup, pipeline_task.PipelineTask]

Expand Down Expand Up @@ -462,7 +457,7 @@ def get_outputs_for_all_groups(
group_name_to_group = {group.name: group for group in all_groups}
group_name_to_children = {
group.name: [group.name for group in group.groups] +
[task.name for task in group.tasks] for group in all_groups
[task.name for task in group.tasks] for group in all_groups
}

outputs = collections.defaultdict(dict)
Expand Down Expand Up @@ -762,7 +757,9 @@ def get_dependencies(
# then make this validation dsl.Collected-aware
elif isinstance(upstream_parent_group, tasks_group.ParallelFor):
upstream_tasks_that_downstream_consumers_from = [
channel.task.name for channel in task._channel_inputs if channel.task
channel.task.name
for channel in task._channel_inputs
if channel.task
]
has_data_exchange = upstream_task.name in upstream_tasks_that_downstream_consumers_from
# don't raise for .after
Expand Down

0 comments on commit 663dd90

Please sign in to comment.