Skip to content

Commit

Permalink
Add component type as a parameter pass to BCLExecutorOperator
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 662595884
  • Loading branch information
tfx-copybara committed Aug 13, 2024
1 parent 90c8da3 commit 0082555
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions tfx/orchestration/portable/launcher.py
Original file line number Diff line number Diff line change
Expand Up @@ -193,9 +193,15 @@ def __init__(
self._driver_operators.update(custom_driver_operators or {})

self._executor_operator = None
# redundant line for external usage.
executor_operator = None
if executor_spec:
self._executor_operator = self._executor_operators[type(executor_spec)](
executor_spec, platform_config)
if executor_operator is None:
executor_operator = self._executor_operators[type(executor_spec)](
executor_spec=executor_spec, platform_config=platform_config
)
self._executor_operator = executor_operator

self._output_resolver = outputs_utils.OutputsResolver(
pipeline_node=self._pipeline_node,
pipeline_info=self._pipeline_info,
Expand Down

0 comments on commit 0082555

Please sign in to comment.