def _extract_augmentor_param(augmentor_yaml):
    """ Takes augmentor yaml dictionary and determines name and parameters
    Args:
      augmentor_yaml: Parameter dictionary containing augmentor data.
    """

    if isinstance(augmentor_yaml, str):
        augmentor_name = augmentor_yaml
        augmentor_param = None
    elif isinstance(augmentor_yaml, dict):
        assert len(augmentor_yaml) == 1
        augmentor_name, augmentor_param = next(iter(augmentor_yaml.items()))
    else:
        raise RuntimeError(
            'Augmentor parameters of type {0} not supported'.format(
                type(augmentor_yaml)))
    return augmentor_name, augmentor_param