def download_all(download_config, scenario): """ Download all datasets for a scenario or requested datset to cache. """ def _print_scenario_names(): logger.info( f"The following scenarios are available based upon config file {download_config}:" ) for scenario in config["scenario"].keys(): logger.info(scenario) config = _read_validate_scenario_config(download_config) if scenario == "all": for scenario in config["scenario"].keys(): for dataset in config["scenario"][scenario]["dataset_name"]: _download_data(dataset) elif scenario == "list": _print_scenario_names() else: if scenario not in config["scenario"].keys(): logger.info(f"The scenario name {scenario} is not valid.") _print_scenario_names() raise ValueError("Invalid scenario name.") for dataset in config["scenario"][scenario]["dataset_name"]: _download_data(dataset)
def download_all(download_config, scenario): config = _read_validate_scenario_config(download_config) if scenario == "all": for scenario in config["scenario"].keys(): for weights_file in config["scenario"][scenario]["weights_file"]: _download_weights(weights_file) elif scenario == "list": return else: for weights_file in config["scenario"][scenario]["weights_file"]: _download_weights(weights_file)