From 46cd12e1e5c4fb1879e39ae4968a82a9c95fcf92 Mon Sep 17 00:00:00 2001
From: David Luevano <55825613+luevano@users.noreply.github.com>
Date: Thu, 12 Dec 2019 03:32:27 -0700
Subject: Reflect added features

---
 main.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'main.py')

diff --git a/main.py b/main.py
index a7e608644..679bb4eea 100644
--- a/main.py
+++ b/main.py
@@ -92,9 +92,9 @@ printc('\tL-J matrices calculation took {:.4f} seconds.'.format(toc-tic),
        Fore.GREEN)
 
 # Coulomb Matrix ML
-do_ml(cm_data, 'CM', energy_pbe0, 1000, 100, 1000.0)
+do_ml(cm_data, energy_pbe0, 1000, 100, sigma=1000.0, desc_type='CM')
 # Lennard-Jones Matrix ML
-do_ml(ljm_data, 'L-JM', energy_pbe0, 1000, 100, 1000.0)
+do_ml(ljm_data, energy_pbe0, 1000, 100, sigma=1000.0, desc_type='L-JM')
 
 # End of program
 end_time = time.perf_counter()
-- 
cgit v1.2.3-54-g00ecf