From b983e0ba919e1d5ec12d2f02dad83ef640474d09 Mon Sep 17 00:00:00 2001 From: Ingmar Schuster Date: Sat, 27 May 2023 18:20:05 +0200 Subject: [PATCH] Fixed test Test case had to be adapted Signed-off-by: Ingmar Schuster --- tests/test_kernels/test_stationary.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_kernels/test_stationary.py b/tests/test_kernels/test_stationary.py index 7091e891..2a48cd4d 100644 --- a/tests/test_kernels/test_stationary.py +++ b/tests/test_kernels/test_stationary.py @@ -108,7 +108,7 @@ def test_initialization(self, fields: dict, dim: int) -> None: if field in ["variance", "lengthscale", "period", "alpha"]: assert isinstance(meta[field]["bijector"], tfb.Softplus) if field in ["power"]: - assert isinstance(meta[field]["bijector"], tfb.Identity) + assert isinstance(meta[field]["bijector"], tfb.Sigmoid) # Trainability state assert meta[field]["trainable"] is True @@ -225,7 +225,7 @@ class TestPeriodic(BaseTestKernel): class TestPoweredExponential(BaseTestKernel): kernel = PoweredExponential fields = prod( - {"lengthscale": [0.1, 1.0], "variance": [0.1, 1.0], "power": [0.1, 2.0]} + {"lengthscale": [0.1, 1.0], "variance": [0.1, 1.0], "power": [0.1, 0.9]} ) params = {"test_initialization": fields} default_compute_engine = DenseKernelComputation