From 297f0d59f02f19b697bca6125687aa96f1c2dce6 Mon Sep 17 00:00:00 2001 From: QuanyiLi Date: Mon, 8 May 2023 10:25:22 +0100 Subject: [PATCH] use is_scenario_file to determine --- scenarionet/builder/utils.py | 11 +---------- 1 file changed, 1 insertion(+), 10 deletions(-) diff --git a/scenarionet/builder/utils.py b/scenarionet/builder/utils.py index 5e871cb..4ecf797 100644 --- a/scenarionet/builder/utils.py +++ b/scenarionet/builder/utils.py @@ -25,22 +25,13 @@ def try_generating_summary(file_folder): files = os.listdir(file_folder) summary = {} for file in files: - if file != ScenarioDescription.DATASET.SUMMARY_FILE and file != ScenarioDescription.DATASET.MAPPING_FILE: + if ScenarioDescription.is_scenario_file(file): with open(osp.join(file_folder, file), "rb+") as f: scenario = pickle.load(f) summary[file] = copy.deepcopy(scenario[ScenarioDescription.METADATA]) return summary -def try_generating_mapping(file_folder): - # Create a fake one - files = os.listdir(file_folder) - mapping = {} - for file in files: - mapping[file] = "" - return mapping - - def combine_multiple_dataset( output_path, *dataset_paths, force_overwrite=False, try_generate_missing_file=True, filters: List[Callable] = None ):