import yaml


def load_yaml(yaml_path):
  """Load the yaml|json file

  Returns:
      A dict which contains the config info
  """
  with open(yaml_path, 'r') as f_config:
    config_dict = yaml.full_load(f_config)
  return config_dict