summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--ml_exp/compound.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/ml_exp/compound.py b/ml_exp/compound.py
index f41710588..4c5b21584 100644
--- a/ml_exp/compound.py
+++ b/ml_exp/compound.py
@@ -197,6 +197,8 @@ class Compound:
self.qm9Mulliken = np.empty(self.n, dtype=np.float64)
for i, atom in enumerate(lines[2:self.n + 2]):
+ if db == 'qm9':
+ atom = atom.replace('*^', 'e')
atom_d = atom.split()
self.atoms.append(atom_d[0])