code: def set_image(
      config: Dict[Any, Any], cpu_image: str, gpu_image: str
  ) -> Dict[Any, Any]:
      config = config.copy()
      config.setdefault("environment", {})
      config["environment"]["image"] = {
          "cpu": cpu_image,
          "gpu": gpu_image,
      }
      return config
  
imports: import os
  from pathlib import Path
  from typing import Any, Dict, List, Union
  from determined.common import api, util
  from determined.common.api import authentication
label_desc: Use image node_id 18 for calling a global function with example usage: set_image(config, cpu_image, gpu_image) and returns: config