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: 660981441
  • Loading branch information
tfx-copybara committed Aug 8, 2024
1 parent 3db64bc commit c58ce01
Showing 1 changed file with 14 additions and 2 deletions.
16 changes: 14 additions & 2 deletions tfx/orchestration/portable/launcher.py
Original file line number Diff line number Diff line change
Expand Up @@ -194,8 +194,20 @@ def __init__(

self._executor_operator = None
if executor_spec:
self._executor_operator = self._executor_operators[type(executor_spec)](
executor_spec, platform_config)
if issubclass(
self._executor_operators[type(executor_spec)],
bcl_executor_operator.BCLExecutorOperator,
):
self._executor_operator = self._executor_operators[type(executor_spec)](
executor_spec=executor_spec,
component_type=shared_constants.FormFactor.BCL,
platform_config=platform_config,
)
else:
self._executor_operator = self._executor_operators[type(executor_spec)](
executor_spec=executor_spec, platform_config=platform_config
)

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

0 comments on commit c58ce01

Please sign in to comment.