From a62fa84e284b4c331f58d8062faa9bd74ed3f063 Mon Sep 17 00:00:00 2001
From: David Luevano <55825613+luevano@users.noreply.github.com>
Date: Thu, 12 Dec 2019 04:40:48 -0700
Subject: Reformatting

---
 main.py | 15 +++++----------
 1 file changed, 5 insertions(+), 10 deletions(-)

diff --git a/main.py b/main.py
index d00e4157e..8bea40c96 100644
--- a/main.py
+++ b/main.py
@@ -34,27 +34,22 @@ from do_ml import do_ml
 # Initialization time.
 init_time = time.perf_counter()
 
-# Move to data folder.
+# Data reading.
+tic = time.perf_counter()
+printc('Data reading started.', 'CYAN')
+
 init_path = os.getcwd()
 os.chdir('data')
 data_path = os.getcwd()
 
-printc('Data reading started.', 'CYAN')
-tic = time.perf_counter()
-
-# Read nuclear charge data.
 zi_data = read_nc_data(data_path)
-
-# Read molecule data.
 molecules, nuclear_charge, energy_pbe0, energy_delta = \
     read_db_edata(zi_data, data_path)
 
+os.chdir(init_path)
 toc = time.perf_counter()
 printc('\tData reading took {:.4f} seconds.'.format(toc-tic), 'GREEN')
 
-# Go back to main folder.
-os.chdir(init_path)
-
 # Matrices calculation.
 cm_data = c_matrix_multiple(molecules, nuclear_charge, as_eig=True)
 ljm_data = lj_matrix_multiple(molecules, nuclear_charge, as_eig=True)
-- 
cgit v1.2.3-70-g09d2