Skip to content

Commit 0082555

Browse files
committed
Add component type as a parameter pass to BCLExecutorOperator
PiperOrigin-RevId: 662595884
1 parent 90c8da3 commit 0082555

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

tfx/orchestration/portable/launcher.py

+8-2
Original file line numberDiff line numberDiff line change
@@ -193,9 +193,15 @@ def __init__(
193193
self._driver_operators.update(custom_driver_operators or {})
194194

195195
self._executor_operator = None
196+
# redundant line for external usage.
197+
executor_operator = None
196198
if executor_spec:
197-
self._executor_operator = self._executor_operators[type(executor_spec)](
198-
executor_spec, platform_config)
199+
if executor_operator is None:
200+
executor_operator = self._executor_operators[type(executor_spec)](
201+
executor_spec=executor_spec, platform_config=platform_config
202+
)
203+
self._executor_operator = executor_operator
204+
199205
self._output_resolver = outputs_utils.OutputsResolver(
200206
pipeline_node=self._pipeline_node,
201207
pipeline_info=self._pipeline_info,

0 commit comments

Comments
 (0)