fix: imports for hf hub errors

amitkparekh · Oct 9, 2024 · 316a3a0 · 316a3a0
1 parent d92b3f5
commit 316a3a0
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 8 deletions.
diff --git a/pdm.lock b/pdm.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -47,6 +47,7 @@ dependencies = [
     "moviepy>=1.0.3",
     "more-itertools>=10.2.0",
     "polars>=0.20.6",
+    "huggingface-hub>=0.25.2",
 ]
 
 [tool.pdm]

diff --git a/src/cogelot/common/hf_datasets.py b/src/cogelot/common/hf_datasets.py
@@ -10,10 +10,7 @@
 from datasets.features.features import require_decoding
 from datasets.table import embed_table_storage
 from huggingface_hub import HfApi, snapshot_download
-from huggingface_hub.utils._errors import (
-    BadRequestError,
-    HfHubHTTPError,
-)
+from huggingface_hub.errors import BadRequestError, HfHubHTTPError
 from loguru import logger
 from tqdm import tqdm