warnings.filterwarnings

Lightning-AI · Mar 14, 2024 · 56da0e2 · 56da0e2
1 parent 9e43830
commit 56da0e2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/unittests/__init__.py b/tests/unittests/__init__.py
@@ -1,4 +1,5 @@
 import os.path
+import warnings
 from typing import NamedTuple
 
 import numpy
@@ -28,6 +29,10 @@
 
 _reference_cachier = cachier(cache_dir=_PATH_TEST_CACHE, separate_files=True)
 
+# ignore FutureWarnings while testing (mainly appearing with DDP runs)
+warnings.filterwarnings("ignore", category=FutureWarning, module="sklearn.*")
+warnings.filterwarnings("ignore", category=FutureWarning, module="transformers.*")
+
 
 if torch.cuda.is_available():
     torch.backends.cuda.matmul.allow_tf32 = False