From db64425a5580a49312e313a6e75e7a296eb93b35 Mon Sep 17 00:00:00 2001
From: David Luevano <55825613+luevano@users.noreply.github.com>
Date: Mon, 23 Dec 2019 12:23:46 -0700
Subject: Restructure code and bug fix

---
 lj_matrix/lj_matrix.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'lj_matrix/lj_matrix.py')

diff --git a/lj_matrix/lj_matrix.py b/lj_matrix/lj_matrix.py
index 2a56a3cdf..0c16b5686 100644
--- a/lj_matrix/lj_matrix.py
+++ b/lj_matrix/lj_matrix.py
@@ -38,6 +38,8 @@ def lj_matrix(mol_data,
     Creates the Lennard-Jones Matrix from the molecule data given.
     mol_data: molecule data, matrix of atom coordinates.
     nc_data: nuclear charge data, array of atom data.
+    sigma: sigma value.
+    epsilon: epsilon value.
     max_len: maximum amount of atoms in molecule.
     as_eig: if data should be returned as matrix or array of eigenvalues.
     bohr_radius_units: if units should be in bohr's radius units.
@@ -171,8 +173,8 @@ def lj_matrix(mol_data,
 def lj_matrix_multiple(mol_data,
                        nc_data,
                        pipe=None,
-                       sigma=1,
-                       epsilon=1,
+                       sigma=1.0,
+                       epsilon=1.0,
                        max_len=25,
                        as_eig=True,
                        bohr_radius_units=False):
-- 
cgit v1.2.3-54-g00ecf