Reformatting

1 files 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)