diff options
Diffstat (limited to 'ml_exp/qm7db.py')
-rw-r--r-- | ml_exp/qm7db.py | 6 |
1 files changed, 3 insertions, 3 deletions
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) |