Skip to content

Commit

Permalink
Fixed test
Browse files Browse the repository at this point in the history
Test case had to be adapted

Signed-off-by: Ingmar Schuster <ingmar.schuster@gmail.com>
  • Loading branch information
ingmarschuster authored May 27, 2023
1 parent 9fa0e6e commit b983e0b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/test_kernels/test_stationary.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit b983e0b

Please sign in to comment.