diff --git a/mlatom/interfaces/torchani_interface.py b/mlatom/interfaces/torchani_interface.py index 445abc0..7f02e8b 100644 --- a/mlatom/interfaces/torchani_interface.py +++ b/mlatom/interfaces/torchani_interface.py @@ -1934,7 +1934,7 @@ def parse_ani1xnr_resources(): if not os.path.exists(local_dir+'ani-1xnr-main'): os.makedirs(local_dir, exist_ok=True) print(f'Downloading ANI-1xnr model parameters ...') - resource_res = requests.get(url) + resource_res = requests.get(url, timeout=30) resource_zip = zipfile.ZipFile(io.BytesIO(resource_res.content)) resource_zip.extractall(local_dir) return local_dir diff --git a/mlatom/model_cls.py b/mlatom/model_cls.py index 99177db..d9f89f7 100644 --- a/mlatom/model_cls.py +++ b/mlatom/model_cls.py @@ -284,7 +284,7 @@ def _download(self, link=None, headers=None, target=None): assert target is not None, "Please provide target directory to be downloaded" print(f'Start downloading model from {link} to {target}'); sys.stdout.flush() try: - response = requests.get(link, headers=headers, stream=True, allow_redirects=True) + response = requests.get(link, headers=headers, stream=True, allow_redirects=True, timeout=30) total_size = int(response.headers.get("content-length", 0)) target += '.temp'