Update imports

ziatdinovmax · Mar 1, 2024 · f3289d8 · f3289d8
1 parent 02f3001
commit f3289d8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/gpax/models/__init__.py b/gpax/models/__init__.py
@@ -15,6 +15,7 @@
 from .linreg import LinReg
 from .sparse_gp import viSparseGP
 from .bnn import BNN
+from .vibnn import viBNN
 
 __all__ = [
     "ExactGP",
@@ -33,5 +34,6 @@
     "LinReg",
     "MeasuredNoiseGP",
     "viSparseGP",
-    "BNN"
+    "BNN",
+    "viBNN"
 ]