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

---
 ml_exp/__init__.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/ml_exp/__init__.py b/ml_exp/__init__.py
index b40957baf..e1d3bcd6f 100644
--- a/ml_exp/__init__.py
+++ b/ml_exp/__init__.py
@@ -25,7 +25,7 @@ from ml_exp.representations import coulomb_matrix, lennard_jones_matrix,\
         get_helping_data, adjacency_matrix, epsilon_index, check_bond, bag_of_bonds
 from ml_exp.readdb import qm7db, qm9db
 from ml_exp.data import NUCLEAR_CHARGE, POSSIBLE_BONDS
-from ml_exp.kernels import laplauss_kernel
+from ml_exp.kernels import gaussian_kernel, laplauss_kernel, wasserstein_kernel
 from ml_exp.krr import krr, multi_krr
 
 __all__ = ['Compound',
@@ -38,7 +38,9 @@ __all__ = ['Compound',
            'bag_of_bonds',
            'qm7db',
            'qm9db',
-           'laplauss_kernel',
+           'gaussian_kernel',
+           'laplacian_kernel',
+           'wasserstein_kernel',
            'krr',
            'multi_krr',
            'NUCLEAR_CHARGE',
-- 
cgit v1.2.3-54-g00ecf