diff options
Diffstat (limited to 'ml_exp/compound.py')
-rw-r--r-- | ml_exp/compound.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ml_exp/compound.py b/ml_exp/compound.py index 0a8b89610..9c7102860 100644 --- a/ml_exp/compound.py +++ b/ml_exp/compound.py @@ -36,6 +36,7 @@ class Compound: self.name = None self.n = None + self.extra = None # In case a comment is in the compound file. self.atoms = None self.atoms_nc = None self.coordinates = None @@ -139,6 +140,7 @@ class Compound: self.name = filename.split('/')[-1] self.n = int(lines[0]) + self.extra = lines[1] self.atoms = [] self.atoms_nc = np.empty(self.n, dtype=int) self.coordinates = np.empty((self.n, 3), dtype=float) |