do not use urllib; use requests for all network things

SBU-BMI · May 30, 2023 · eacd806 · eacd806
1 parent 00cce49
commit eacd806
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/wsinfer_zoo/client.py b/wsinfer_zoo/client.py
@@ -5,8 +5,6 @@
 from datetime import datetime
 from pathlib import Path
 from typing import Dict, List, Optional, Sequence, Tuple
-import urllib.error
-import urllib.request
 
 import requests
 from huggingface_hub import hf_hub_download
@@ -222,8 +220,7 @@ def _download_registry_if_necessary():
     """
     try:
         if _remote_registry_is_newer():
-            urllib.request.urlretrieve(
-                WSINFER_ZOO_REGISTRY_URL, WSINFER_ZOO_REGISTRY_DEFAULT_PATH
-            )
-    except (requests.RequestException, urllib.error.URLError) as e:
+            r = requests.get(WSINFER_ZOO_REGISTRY_URL)
+            WSINFER_ZOO_REGISTRY_DEFAULT_PATH.write_bytes(r.content)
+    except requests.RequestException as e:
         print(f"Could not download most recent registry, error: {e}")