diff --git a/scenes/download.py b/scenes/download.py index 6370b9c1b6351b2dd259bbeb0ffe051f55e3f571..8ed59b41c5b6c36721f10e1e8a01ade7d9c796d9 100644 --- a/scenes/download.py +++ b/scenes/download.py @@ -16,7 +16,7 @@ def compute_md5(fname): """ hash_md5 = hashlib.md5() with open(fname, "rb") as f: - for chunk in tqdm.tqdm(iter(lambda: f.read(4096), b"")): + for chunk in iter(lambda: f.read(4096), b""): hash_md5.update(chunk) return hash_md5.hexdigest() diff --git a/test/download_test.py b/test/download_test.py index 0f1e645a8835891991007c2d82db3e283aced6fe..b3eaf11ad5f21b0fcb434c7ec9d5c6fa49b902ac 100644 --- a/test/download_test.py +++ b/test/download_test.py @@ -7,10 +7,13 @@ parser = argparse.ArgumentParser(description="Download test",) parser.add_argument("--refimage", required=True) parser.add_argument("--theia_cfg", required=True) parser.add_argument("--download_dir") +parser.add_argument("--year", type=int, default=2020) +parser.add_argument("--month", type=int, default=1) +parser.add_argument("--day", type=int, default=1) params = parser.parse_args() # Get all scenes in the root_dir _, _, bbox = utils.get_epsg_extent_bbox(params.refimage) -acq_date = datetime.datetime(year=2020, month=1, day=1) +acq_date = datetime.datetime(year=params.year, month=params.month, day=params.day) download.download_closest(config_file=params.theia_cfg, bbox_wgs84=bbox, acq_date=acq_date, download_dir=params.download_dir)