summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--ml_exp/krr.py3
1 files changed, 2 insertions, 1 deletions
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