diff options
Diffstat (limited to 'ml_exp/__init__.py')
| -rw-r--r-- | ml_exp/__init__.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/ml_exp/__init__.py b/ml_exp/__init__.py index d81f0999a..dcb10a1df 100644 --- a/ml_exp/__init__.py +++ b/ml_exp/__init__.py @@ -22,7 +22,7 @@ SOFTWARE. """ from ml_exp.compound import Compound from ml_exp.representations import coulomb_matrix, lennard_jones_matrix,\ - first_neighbor_matrix, adjacency_matrix + first_neighbor_matrix, adjacency_matrix, check_bond, bag_of_stuff # If somebody does "from package import *", this is what they will # be able to access: @@ -30,4 +30,6 @@ __all__ = ['Compound', 'coulomb_matrix', 'lennard_jones_matrix', 'first_neighbor_matrix', - 'adjacency_matrix'] + 'adjacency_matrix', + 'check_bond', + 'bag_of_stuff'] |