Update imports

ziatdinovmax · Oct 7, 2023 · 7e2a2dd · 7e2a2dd
1 parent 308833a
commit 7e2a2dd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gpax/__init__.py b/gpax/__init__.py
@@ -4,8 +4,8 @@
 from . import acquisition
 from .hypo import sample_next
 from .models import (DKL, CoregGP, ExactGP, MultiTaskGP, iBNN, vExactGP,
-                     vi_iBNN, viDKL, viGP, viMTDKL)
+                     vi_iBNN, viDKL, viGP, viMTDKL, VarNoiseGP)
 
 __all__ = ["utils", "kernels", "mtkernels", "acquisition", "ExactGP", "vExactGP", "DKL",
-           "viDKL", "iBNN", "vi_iBNN", "MultiTaskGP", "viMTDKL", "viGP", "sPM",
+           "viDKL", "iBNN", "vi_iBNN", "MultiTaskGP", "viMTDKL", "viGP", "sPM", "VarNoiseGP",
            "CoregGP", "sample_next", "__version__"]