code: def get_node_id(node_id: int):
      """
      If using SLURM, we get environment variables like SLURMD_NODENAME,
      SLURM_NODEID to get information about the current node.
      Useful to set the node_id automatically.
      """
      node_list = os.environ.get("SLURM_STEP_NODELIST")
      if node_list is not None:
          node_name = str(os.environ["SLURMD_NODENAME"])
          node_id = int(os.environ.get("SLURM_NODEID"))
          logging.info(
              f"SLURM job: node_name: {node_name}, node_id: {node_id}"
          )
      return node_id
  
imports: import logging
  import os
label_desc: Use image node_id 1 for calling a global function with example usage: get_node_id(node_id) and returns: node_id