From a0e73407bfef3569a994c07d5bb9c8df97518e30 Mon Sep 17 00:00:00 2001
From: David Luevano Alvarado <55825613+luevano@users.noreply.github.com>
Date: Wed, 26 Feb 2020 05:32:35 -0700
Subject: Add feature

---
 ml_exp/compound.py | 2 ++
 ml_exp/qm7db.py    | 6 +++---
 2 files changed, 5 insertions(+), 3 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)
diff --git a/ml_exp/qm7db.py b/ml_exp/qm7db.py
index f9950c317..8d2453f45 100644
--- a/ml_exp/qm7db.py
+++ b/ml_exp/qm7db.py
@@ -49,9 +49,9 @@ def qm7db(nc,
         compounds[i].pbe0 = float(line[1])
         compounds[i].delta = float(line[1]) - float(line[2])
 
-    # Shuffle the compounds list
-    random.seed(r_seed)
-    random.shuffle(compounds)
+    if is_shuffled:
+        random.seed(r_seed)
+        random.shuffle(compounds)
 
     e_pbe0 = np.array([compound.pbe0 for compound in compounds], dtype=float)
     e_delta = np.array([compound.delta for compound in compounds], dtype=float)
-- 
cgit v1.2.3-54-g00ecf