code: def _get_pipeline_node(
      pipeline: pipeline_pb2.Pipeline, node_id: str
  ) -> pipeline_pb2.PipelineNode:
      """Gets corresponding pipeline node from IR given the node_id."""
      for node in pipeline.nodes:
          if node.pipeline_node and (
              node.pipeline_node.node_info.id == node_id
          ):
              return node.pipeline_node
      raise status_lib.StatusNotOkError(
          code=status_lib.Code.INVALID_ARGUMENT,
          message=f"Failed to find corresponding node in the IR, node id: {node_id}",
      )
  
file_path: tfx/tfx/orchestration/experimental/centralized_kubernetes_orchestrator/kubernetes_task_scheduler.py
imports: import threading
  from tfx.orchestration import data_types_utils
  from tfx.orchestration import metadata
  from tfx.orchestration.experimental.centralized_kubernetes_orchestrator import kubernetes_job_runner
  from tfx.orchestration.experimental.core import task
  from tfx.orchestration.experimental.core import task_scheduler
  from tfx.orchestration.portable import data_types
  from tfx.proto.orchestration import executable_spec_pb2
  from tfx.proto.orchestration import execution_invocation_pb2
  from tfx.proto.orchestration import pipeline_pb2
  from tfx.utils import status
label_desc: Use image node_id 2 for calling a global function with example usage: _get_pipeline_node(pipeline, node_id) and returns: node