test(tests/test_kernels/test_nonstationary): abs

JaxGaussianProcesses · Jul 1, 2024 · e697365 · e697365
1 parent b3e603c
commit e697365
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_kernels/test_nonstationary.py b/tests/test_kernels/test_nonstationary.py
@@ -241,4 +241,4 @@ def test_values_by_monte_carlo_in_special_case(self, order: int) -> None:
         integrands = H_a * H_b * (weights_a**order) * (weights_b**order)
         Kab_approx = 2.0 * jnp.mean(integrands)
 
-        assert jnp.max(Kab_approx - Kab_exact) < 1e-4
+        assert jnp.max(jnp.abs(Kab_approx - Kab_exact)) < 1e-4