From 2aca095533d6b51b32e0efeb0eaff300593c3a08 Mon Sep 17 00:00:00 2001 From: Robert Shelton Date: Mon, 18 Aug 2025 11:54:01 -0400 Subject: [PATCH] disable progress bar by default --- redisvl/utils/vectorize/text/huggingface.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/redisvl/utils/vectorize/text/huggingface.py b/redisvl/utils/vectorize/text/huggingface.py index 2188d9e4..ac13b08e 100644 --- a/redisvl/utils/vectorize/text/huggingface.py +++ b/redisvl/utils/vectorize/text/huggingface.py @@ -128,6 +128,10 @@ def _embed(self, text: str, **kwargs) -> List[float]: if not isinstance(text, str): raise TypeError("Must pass in a str value to embed.") + if "show_progress_bar" not in kwargs: + # disable annoying tqdm by default + kwargs["show_progress_bar"] = False + embedding = self._client.encode([text], **kwargs)[0] return embedding.tolist() @@ -151,6 +155,9 @@ def _embed_many( raise TypeError("Must pass in a list of str values to embed.") if len(texts) > 0 and not isinstance(texts[0], str): raise TypeError("Must pass in a list of str values to embed.") + if "show_progress_bar" not in kwargs: + # disable annoying tqdm by default + kwargs["show_progress_bar"] = False embeddings: List = [] for batch in self.batchify(texts, batch_size, None):