diff options
author | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-02-26 05:32:35 -0700 |
---|---|---|
committer | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-02-26 05:32:35 -0700 |
commit | a0e73407bfef3569a994c07d5bb9c8df97518e30 (patch) | |
tree | 3232125c94fdbc3df634f8067f51965792e22c11 /ml_exp/compound.py | |
parent | 5a85580b5f99484b957b65326ec5a19ab627d801 (diff) |
Add feature
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) |