Update imports

ziatdinovmax · Feb 6, 2024 · ce93965 · ce93965
1 parent b438d06
commit ce93965
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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__"]