lazy import huggingface_hub

jainanisha90 · Mar 28, 2024 · cbafc83 · cbafc83
1 parent 8d49821
commit cbafc83
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/fiftyone/utils/huggingface.py b/fiftyone/utils/huggingface.py
@@ -14,7 +14,6 @@
 import requests
 import shutil
 
-import huggingface_hub as hfh
 import yaml
 
 import fiftyone as fo
@@ -27,6 +26,10 @@
 import fiftyone.core.utils as fou
 import fiftyone.types as fot
 
+hfh = fou.lazy_import(
+    "huggingface_hub", callback=lambda: fou.ensure_package("huggingface_hub")
+)
+
 DATASETS_SERVER_URL = "https://datasets-server.huggingface.co"
 DEFAULT_MEDIA_TYPE = "image"
 DATASET_METADATA_FILENAMES = ("fiftyone.yml", "fiftyone.yaml")