Adding retries to the download_dataset method