Edit bag of stuff function name

1 files changed, 5 insertions, 5 deletions

diff --git a/ml_exp/representations.py b/ml_exp/representations.py
index a85fee11f..1fe55aa5f 100644
--- a/ml_exp/representations.py
+++ b/ml_exp/representations.py
@@ -308,15 +308,15 @@ def check_bond(bags,
     return False, None
 
 
-def bag_of(cm,
-           atoms,
-           stuff='bonds',
-           size=23):
+def bag_of_stuff(cm,
+                 atoms,
+                 stuff='bonds',
+                 size=23):
     """
     Creates the Bag of Bonds using the Coulomb Matrix.
     cm: coulomb matrix.
     atoms: list of atoms.
-    size: maximum amount of atoms.
+    size: compound size.
     """
     if cm is None:
         raise ValueError('Coulomb Matrix hasn\'t been initialized for the \