def get_dataset_configs()

in src/fmeval/eval_algorithms/util.py [0:0]

• 9 lines of code
• 6 McCabe index (conditional complexity)

def get_dataset_configs(data_config: Optional[Union[DataConfig, List[DataConfig]]], eval_name: str) -> List[DataConfig]:
    if not data_config:
        return [DATASET_CONFIGS[dataset_name] for dataset_name in EVAL_DATASETS[eval_name]]
    elif isinstance(data_config, list):
        return data_config
    elif isinstance(data_config, tuple):
        return [cfg for cfg in data_config]
    else:
        return [data_config]