download: handle timeout

agimus-project · Dec 14, 2023 · 223ee4b · 223ee4b
1 parent fe82583
commit 223ee4b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/happypose/toolbox/utils/download.py b/happypose/toolbox/utils/download.py
@@ -369,7 +369,7 @@ async def adownload(self, download_path, local_path, flags=None):
     async def download_dir(self, download_path, local_path, flags):
         try:
             r = await self.client.get(download_path)
-        except httpx.PoolTimeout:
+        except (httpx.PoolTimeout, httpx.ReadTimeout):
             logger.error(f"Failed {download_path} with timeout")
             return
         if r.status_code != 200: