From 2e2b4cfd06db1738491d83ecb6ccc66655ff78e2 Mon Sep 17 00:00:00 2001
From: Maxim Ziatdinov <34245227+ziatdinovmax@users.noreply.github.com>
Date: Fri, 5 Jan 2024 13:03:05 -0800
Subject: [PATCH] Fix imports

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

diff --git a/gpax/models/__init__.py b/gpax/models/__init__.py
index a87f348..186b483 100644
--- a/gpax/models/__init__.py
+++ b/gpax/models/__init__.py
@@ -10,7 +10,7 @@
 from .vi_mtdkl import viMTDKL
 from .mtgp import MultiTaskGP
 from .corgp import CoregGP
-from uigp import UIGP
+from .uigp import UIGP
 
 __all__ = [
     "ExactGP",