fix failing tests

JaxGaussianProcesses · Dec 17, 2022 · e0439b3 · e0439b3
1 parent 33013cb
commit e0439b3
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 14 deletions.
diff --git a/gpjax/kernels.py b/gpjax/kernels.py
@@ -428,9 +428,6 @@ def __call__(
         )
 
 
-@deprecation.deprecated(
-    deprecated_in="0.5.5", removed_in="0.6.0", details="Use JaxKern for the SumKernel"
-)
 class SumKernel(CombinationKernel):
     """A kernel that is the sum of a set of kernels."""
 
@@ -449,11 +446,6 @@ def __init__(
         self.combination_fn: Optional[Callable] = jnp.sum
 
 
-@deprecation.deprecated(
-    deprecated_in="0.5.5",
-    removed_in="0.6.0",
-    details="Use JaxKern for the ProductKernel",
-)
 class ProductKernel(CombinationKernel):
     """A kernel that is the product of a set of kernels."""
 
@@ -962,11 +954,6 @@ def _initialise_params(self, key: PRNGKeyType) -> Dict:
 ##########################################
 # Graph kernels
 ##########################################
-@deprecation.deprecated(
-    deprecated_in="0.5.5",
-    removed_in="0.6.0",
-    details="Use JaxKern for the EigenKernelComputation",
-)
 class EigenKernelComputation(AbstractKernelComputation):
     def __init__(
         self,

diff --git a/tests/test_kernels.py b/tests/test_kernels.py
@@ -558,7 +558,6 @@ def test_graph_kernel():
 
     # Create graph kernel
     kern = GraphKernel(laplacian=L)
-    assert isinstance(kern, GraphKernel)
     assert kern.num_vertex == n_verticies
     assert kern.evals.shape == (n_verticies, 1)
     assert kern.evecs.shape == (n_verticies, n_verticies)