diff options
| -rw-r--r-- | ml_exp/__init__.py | 4 | ||||
| -rw-r--r-- | ml_exp/compound.py | 4 |
2 files changed, 4 insertions, 4 deletions
diff --git a/ml_exp/__init__.py b/ml_exp/__init__.py index 9e64d7754..4d672efd7 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,\ - get_helping_data, adjacency_matrix, check_bond, bag_of_stuff + get_helping_data, adjacency_matrix, check_bond, bag_of_bonds from ml_exp.math import cholesky_solve from ml_exp.qm7db import qm7db from ml_exp.do_ml import simple_ml, do_ml @@ -33,7 +33,7 @@ __all__ = ['Compound', 'get_helping_data', 'adjacency_matrix', 'check_bond', - 'bag_of_stuff', + 'bag_of_bonds', 'cholesky_solve', 'qm7db', 'simple_ml', diff --git a/ml_exp/compound.py b/ml_exp/compound.py index 23c66ec27..ec5085797 100644 --- a/ml_exp/compound.py +++ b/ml_exp/compound.py @@ -23,7 +23,7 @@ SOFTWARE. import numpy as np from ml_exp.data import NUCLEAR_CHARGE from ml_exp.representations import coulomb_matrix, lennard_jones_matrix,\ - get_helping_data, adjacency_matrix, bag_of_stuff + get_helping_data, adjacency_matrix, bag_of_bonds class Compound: @@ -139,7 +139,7 @@ class Compound: Generate the Bag of Stuff for the compound. size: compound size. """ - self.bob = bag_of_stuff(self.cm, + self.bob = bag_of_bonds(self.cm, self.atoms, size=size) |