Beispiel #1
0
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)
Beispiel #2
0
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)