From ccbba69e50248938f9ec88cdc9a8f763c84b2cef Mon Sep 17 00:00:00 2001
From: Maxim Ziatdinov <34245227+ziatdinovmax@users.noreply.github.com>
Date: Sat, 7 Oct 2023 19:28:10 -0400
Subject: [PATCH] Update imports

---
 gpax/models/__init__.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/gpax/models/__init__.py b/gpax/models/__init__.py
index 84f2f9a..29e2d89 100644
--- a/gpax/models/__init__.py
+++ b/gpax/models/__init__.py
@@ -1,6 +1,7 @@
 from .gp import ExactGP
 from .vgp import vExactGP
 from .vigp import viGP
+from hskgp import VarNoiseGP
 from .spm import sPM
 from .ibnn import iBNN
 from .vi_ibnn import vi_iBNN