code: def select_dataset(session_type, conf):
      """CLI snippet for selection the dataset for training."""
  
      if session_type == "training":
          logger.info("Select training set")
          datasets = {}
          for i, data in enumerate(conf["training_sets"]):
              datasets[str(i)] = data
              logger.info("{}: {}".format(i, data))
          dataset = datasets[select_option(datasets)]
  
          return dataset
      else:
          logger.info("Select test set")
          datasets = {}
          for i, data in enumerate(conf["test_sets"]):
              datasets[str(i)] = data
              logger.info("{}: {}".format(i, data))
          dataset = datasets[select_option(datasets)]
  
          return dataset
  
file_path: image-super-resolution/ISR/utils/utils.py
imports: import os
  import argparse
  from datetime import datetime
  import numpy
  import yaml
  from ISR.utils.logger import get_logger
label_desc: Use image node_id 14 for calling a global function with example usage: select_dataset(session_type, conf) and returns: dataset, dataset