diff --git a/test/drs_import.py b/test/drs_import.py index 9aec815b423f7d6e9d56a72d62cf8926b5966648..2be81e5111b24df20dbcaf2ccf358a4ef58c1e22 100644 --- a/test/drs_import.py +++ b/test/drs_import.py @@ -3,12 +3,14 @@ from scenes import drs # Arguments parser = argparse.ArgumentParser(description="Test",) -parser.add_argument("--root_dir", help="Root directory containing MS and PAN folders", required=True) +parser.add_argument("--root_dir", nargs='+', help="Root directories containing MS and PAN folders", required=True) parser.add_argument("--out_pickle", help="Output pickle file", required=True) params = parser.parse_args() # Get all scenes in the root_dir -scenes = drs.get_spot67_scenes(params.root_dir) +scenes = [] +for root_dir in params.root_dir: + scenes += drs.get_spot67_scenes(root_dir) # Save scenes in a pickle file drs.save_scenes(scenes, params.out_pickle)