def _get_workload_container_on_node()

in cli/configcheck.py [0:0]

• 12 lines of code
• 3 McCabe index (conditional complexity)

def _get_workload_container_on_node(node_name: str) -> tuple[str, str] | None:
  """Returns the workload container actively running on a node which requests GPUs."""
  pod_to_container_names = _get_workload_containers_on_node(node_name)
  if not pod_to_container_names:
    return None
  elif len(pod_to_container_names.items()) > 1:
    raise click.Abort(
        f'Multiple workload containers found on node {node_name}:'
        f' {pod_to_container_names.values()}. Please limit to running a single'
        ' workload container per node.'
    )
  pod_name, workload_container = pod_to_container_names.popitem()
  return (pod_name, workload_container)