Add component type as a parameter pass to BCLExecutorOperator

PiperOrigin-RevId: 660981441
tensorflow · Aug 9, 2024 · 42c0525 · 42c0525
1 parent ffb176a
commit 42c0525
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git 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
@@ -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,