code: def get_node_ids() -> Tuple[str, List[str]]:
      """Get the node ids of the head node and a worker node"""
      head_node_id = None
      nodes = list_nodes()
      worker_node_ids = []
      for node in nodes:
          if node.is_head_node:
              head_node_id = node.node_id
          else:
              worker_node_ids += [node.node_id]
      return head_node_id, worker_node_ids
  
imports: import os
  import sys
  import time
  from dataclasses import dataclass
  from typing import Callable, List, Optional, Tuple
  import pytest
  import ray
  import ray._private.ray_constants
  from ray._private.test_utils import wait_for_condition
  from ray.autoscaler.v2.schema import ClusterStatus, LaunchRequest, NodeInfo, ResourceRequestByCount
  from ray.autoscaler.v2.sdk import get_cluster_status, request_cluster_resources
  from ray.autoscaler.v2.tests.util import get_available_resources, get_cluster_resource_state, get_total_resources, report_autoscaling_state
  from ray.core.generated import autoscaler_pb2, autoscaler_pb2_grpc
  from ray.core.generated.autoscaler_pb2 import ClusterResourceState, NodeStatus
  from ray.util.state.api import list_nodes
label_desc: Use image node_id 2 for calling a global function with example usage: get_node_ids() and returns: head_node_id,  worker_node_ids