From 2fcdc24699fc05a867cd6701a362f71bd63d6117 Mon Sep 17 00:00:00 2001 From: Jamie Hardt Date: Wed, 3 Sep 2025 11:16:51 -0700 Subject: [PATCH] Made the warnings filter more specific --- ucsinfer/__main__.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/ucsinfer/__main__.py b/ucsinfer/__main__.py index 9c98514..9ef05b2 100644 --- a/ucsinfer/__main__.py +++ b/ucsinfer/__main__.py @@ -246,7 +246,10 @@ def evaluate(dataset, offset, limit, model, no_foley): if __name__ == '__main__': os.environ['TOKENIZERS_PARALLELISM'] = 'false' + # sentence_transformers generates an error in PyTorch upon loading import warnings - warnings.simplefilter(action='ignore', category=FutureWarning) + warnings.filterwarnings(action='ignore', module='torch', + category=FutureWarning, + message=r"`encoder_attention_mask` is deprecated.*") ucsinfer()