chore: suppress UserWarning from huggingface_hub

mostly-ai · Feb 20, 2025 · 4fbd120 · 4fbd120
1 parent b93d93d
commit 4fbd120
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/mostlyai/qa/__init__.py b/mostlyai/qa/__init__.py
@@ -26,9 +26,7 @@
 __version__ = "1.5.5"
 
 warnings.filterwarnings("ignore", category=FutureWarning, module="phik")
+warnings.filterwarnings("ignore", category=UserWarning, module="huggingface_hub")
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 if Version(pd.__version__) >= Version("2.2.0"):
     pd.set_option("future.no_silent_downcasting", True)
-
-# Suppress HF_TOKEN warning since we only use public models
-warnings.filterwarnings("ignore", category=UserWarning, message="The secret.*HF_TOKEN.*")