ignoring warnings

berylliumsec · Mar 6, 2024 · 2f1ac3a · 2f1ac3a
1 parent 1da48dd
commit 2f1ac3a
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/eclipse/eclipse.py b/src/eclipse/eclipse.py
@@ -20,11 +20,10 @@
 
 # Suppress specific warning from transformers
 warnings.filterwarnings(
-    "ignore",
+    action="ignore",
     message="Special tokens have been added in the vocabulary, make sure the associated word embeddings are fine-tuned or trained.",
+    category=UserWarning
 )
-
-
 # Configure basic logging
 # This will set the log level to ERROR, meaning only error and critical messages will be logged
 # You can specify a filename to write the logs to a file; otherwise, it will log to stderr