code: def get_node_id():
      return ray.get_runtime_context().get_node_id()
  
imports: import ray
  import torch
  from ray._private.test_utils import wait_for_condition
label_desc: Use image node_id 2 for calling a global function with example usage: get_node_id() and returns: ray