diff options
| author | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-03-26 16:36:29 -0700 |
|---|---|---|
| committer | David Luevano Alvarado <55825613+luevano@users.noreply.github.com> | 2020-03-26 16:36:29 -0700 |
| commit | 3a7718fc9bab26cd5c92cc69a8ef9ac27b43d88a (patch) | |
| tree | 0975f0b0478e713623668668de20697b5adff35c | |
| parent | 5097ecede7eea2b0d17e3a653ae282a568d4adfc (diff) | |
Fix flattening
| -rw-r--r-- | ml_exp/compound.py | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/ml_exp/compound.py b/ml_exp/compound.py index fac9015a2..03cfb8519 100644 --- a/ml_exp/compound.py +++ b/ml_exp/compound.py @@ -92,7 +92,8 @@ class Compound: self.nc, size=size, as_eig=as_eig, - bohr_ru=bohr_ru) + bohr_ru=bohr_ru, + flatten=flatten) def gen_ljm(self, diag_value=None, @@ -118,6 +119,7 @@ class Compound: sigma=sigma, epsilon=epsilon, size=size, + flatten=flatten, as_eig=as_eig, bohr_ru=bohr_ru) @@ -139,8 +141,8 @@ class Compound: def gen_am(self, use_forces=False, - flatten=True, - size=23): + size=23, + flatten=True): """ Generate the Adjacency Matrix for the compund. use_forces: if the use of forces instead of k_cx should be used. @@ -151,7 +153,8 @@ class Compound: self.bonds_k, self.bonds_f, use_forces=use_forces, - size=size) + size=size, + flatten=flatten) def gen_ei(self, size=23): |