From 39d7b86cfb578b491c8cbf99905e67f7f7937dc9 Mon Sep 17 00:00:00 2001
From: David Luevano Alvarado <55825613+luevano@users.noreply.github.com>
Date: Tue, 25 Feb 2020 21:13:33 -0700
Subject: Refactor code
---
ml_exp/qm7db.py | 12 +++++-------
1 file changed, 5 insertions(+), 7 deletions(-)
diff --git a/ml_exp/qm7db.py b/ml_exp/qm7db.py
index 4a7beaed6..1f1115ba0 100644
--- a/ml_exp/qm7db.py
+++ b/ml_exp/qm7db.py
@@ -20,7 +20,6 @@ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.
"""
-import os
import numpy as np
import random
@@ -28,19 +27,18 @@ import random
# 'hof_qm7.txt.txt' retrieved from
# https://github.com/qmlcode/tutorial
def qm7db(nc,
- data_path,
+ db_path='data',
r_seed=111):
"""
Creates a list of compounds with the qm7 database.
nc: dictionary containing nuclear charge data.
- data_path: path to the data directory.
+ db_path: path to the database directory.
r_seed: random seed to use for the shuffling.
"""
- os.chdir(data_path)
- fname = 'hof_qm7.txt'
- with open(fname, 'r') as infile:
- lines = infile.readlines()
+ fname = f'{db_path}/hof_qm7.txt'
+ with open(fname, 'r') as f:
+ lines = f.readlines()
# Temporary energy dictionary.
energy_temp = dict()
--
cgit v1.2.3-70-g09d2