Import CatKernel in main namespace

JaxGaussianProcesses · Aug 17, 2023 · b5009ec · b5009ec
1 parent cd9ea9d
commit b5009ec
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/gpjax/__init__.py b/gpjax/__init__.py
@@ -30,6 +30,7 @@
     AbstractKernel,
     BasisFunctionComputation,
     ConstantDiagonalKernelComputation,
+    CatKernel,
     DenseKernelComputation,
     DiagonalKernelComputation,
     EigenKernelComputation,
@@ -112,6 +113,7 @@
     "CollapsedELBO",
     "ELBO",
     "AbstractKernel",
+    "CatKernel",
     "Linear",
     "DenseKernelComputation",
     "DiagonalKernelComputation",