Skip to content

Commit

Permalink
Set _node_execution_options in k8s NodeWrapper
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 564598037
  • Loading branch information
kmonte authored and tfx-copybara committed Sep 12, 2023
1 parent f07d027 commit d8c1ba7
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions tfx/orchestration/experimental/kubernetes/node_wrapper.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,10 @@ def __init__(self, node: base_node.BaseNode):
self._inputs = node.inputs
self._outputs = node.outputs
self._exec_properties = node.exec_properties
# Currently the NodeExecutionOptions in tfx.dsl.experiment.utils is for the
# experimental orchestrator, but we need to set the field here anyways so
# the property can be accessed properly.
self._node_execution_options = None

@property
def type(self) -> str:
Expand Down

0 comments on commit d8c1ba7

Please sign in to comment.