From 42c05250f25cfd99e4b62fa8a715f903164d2a78 Mon Sep 17 00:00:00 2001 From: tfx-team Date: Thu, 8 Aug 2024 14:26:28 -0700 Subject: [PATCH] Add component type as a parameter pass to BCLExecutorOperator PiperOrigin-RevId: 660981441 --- tfx/dependencies.py | 3 --- tfx/orchestration/portable/launcher.py | 6 ++++-- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/tfx/dependencies.py b/tfx/dependencies.py index 24b07a24cf..84098ae7c0 100644 --- a/tfx/dependencies.py +++ b/tfx/dependencies.py @@ -90,9 +90,6 @@ def make_required_install_packages(): 'kubernetes>=10.0.1,<13', 'numpy>=1.16,<2', 'pyarrow>=10,<11', - # TODO: b/358471141 - Orjson 3.10.7 breaks TFX OSS tests. - # Unpin once the issue with installation is resolved. - 'orjson!=3.10.7', # TODO(b/332616741): Scipy version 1.13 breaks the TFX OSS test. # Unpin once the issue is resolved. 'scipy<1.13', diff --git a/tfx/orchestration/portable/launcher.py b/tfx/orchestration/portable/launcher.py index e6de68315e..1c30b9c3a5 100644 --- a/tfx/orchestration/portable/launcher.py +++ b/tfx/orchestration/portable/launcher.py @@ -194,8 +194,10 @@ def __init__( self._executor_operator = None if executor_spec: - self._executor_operator = self._executor_operators[type(executor_spec)]( - executor_spec, platform_config) + 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,