From 1d7ea44f8b5dba01ad2660b8014002b989dff464 Mon Sep 17 00:00:00 2001
From: David Luevano Alvarado <55825613+luevano@users.noreply.github.com>
Date: Thu, 26 Mar 2020 15:53:27 -0700
Subject: Fix kernels

---
 ml_exp/krr.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ml_exp/krr.py b/ml_exp/krr.py
index 60335fa77..4a87884d4 100644
--- a/ml_exp/krr.py
+++ b/ml_exp/krr.py
@@ -30,7 +30,8 @@ except ImportError:
     print('Tensorflow couldn\'t be imported. Maybe it is not installed.')
     TF_AV = False
 from ml_exp.misc import printc
-from ml_exp.kernels import laplauss_kernel
+from ml_exp.kernels import gaussian_kernel, laplacian_kernel,\
+    wasserstein_kernel
 from ml_exp.readdb import qm7db
 
 
-- 
cgit v1.2.3-70-g09d2