diff --git a/gpax/__init__.py b/gpax/__init__.py index f3d7d53..9a1df20 100644 --- a/gpax/__init__.py +++ b/gpax/__init__.py @@ -4,8 +4,9 @@ from . import acquisition from .hypo import sample_next from .models import (DKL, CoregGP, ExactGP, MultiTaskGP, iBNN, vExactGP, - vi_iBNN, viDKL, viGP, viMTDKL, VarNoiseGP, UIGP, MeasuredNoiseGP) + vi_iBNN, viDKL, viGP, viMTDKL, VarNoiseGP, UIGP, + MeasuredNoiseGP, viSparseGP) __all__ = ["utils", "kernels", "mtkernels", "acquisition", "ExactGP", "vExactGP", "DKL", "viDKL", "iBNN", "vi_iBNN", "MultiTaskGP", "viMTDKL", "viGP", "sPM", "VarNoiseGP", - "UIGP", "MeasuredNoiseGP", "CoregGP", "sample_next", "__version__"] + "UIGP", "MeasuredNoiseGP", "viSparseGP", "CoregGP", "sample_next", "__version__"]