filtered more GPy warnings

Helmholtz-AI-Energy · Sep 13, 2024 · 415f963 · 415f963
1 parent 0103fa9
commit 415f963
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/tests/test_surrogate.py b/tests/test_surrogate.py
@@ -10,10 +10,14 @@
 from propulate import Islands, Propulator, surrogate
 from propulate.utils import get_default_propagator, set_logger_config
 
-pytestmark = pytest.mark.filterwarnings(
-    "ignore::DeprecationWarning",
-    match="Assigning the 'data' attribute is an inherently unsafe operation and will be removed in the future.",
-)
+pytestmark = [
+    pytest.mark.filterwarnings(
+        "ignore::DeprecationWarning",
+        match="Assigning the 'data' attribute is an inherently unsafe operation and will be removed in the future.",
+    ),
+    pytest.mark.filterwarnings("ignore::RuntimeWarning", match="invalid value encountered in multiply"),
+    pytest.mark.filterwarnings("ignore::RuntimeWarning", match="overflow encountered in expm1"),
+]
 
 log = logging.getLogger(__name__)  # Get logger instance.
 set_logger_config(level=logging.DEBUG)