diff options
-rw-r--r-- | ml_exp/__init__.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/ml_exp/__init__.py b/ml_exp/__init__.py index 8e803b37e..f5ec7068b 100644 --- a/ml_exp/__init__.py +++ b/ml_exp/__init__.py @@ -21,9 +21,10 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. """ from ml_exp.compound import Compound -from ml_exp.representations import coulomb_matrix +from ml_exp.representations import coulomb_matrix, lennard_jones_matrix # If somebody does "from package import *", this is what they will # be able to access: __all__ = ['Compound', - 'coulomb_matrix'] + 'coulomb_matrix', + 'lennard_jones_matrix'] |