code: def setup_trial_checkpoint_dir():
      checkpoint_dir = os.environ["NNI_OUTPUT_DIR"] + "/checkpoints/"
      pathlib.Path(checkpoint_dir).mkdir(parents=True, exist_ok=True)
      return checkpoint_dir
  
imports: import sys
  import os
  import pathlib
  import logging
  import time
  import argparse
  from collections import namedtuple
  import numpy
  import torch
  import torch.utils.data
  import nni
  from nni.utils import merge_parameter
  from pix2pixlib.data.aligned_dataset import AlignedDataset
  from pix2pixlib.models.pix2pix_model import Pix2PixModel
  from base_params import get_base_params
label_desc: Use image node_id 2 for calling a global function with example usage: setup_trial_checkpoint_dir() and returns: checkpoint_dir