Update imports

Author: ziatdinovmax

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"
 ]