From 3a7718fc9bab26cd5c92cc69a8ef9ac27b43d88a Mon Sep 17 00:00:00 2001
From: David Luevano Alvarado <55825613+luevano@users.noreply.github.com>
Date: Thu, 26 Mar 2020 16:36:29 -0700
Subject: Fix flattening

---
 ml_exp/compound.py | 11 +++++++----
 1 file 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):
-- 
cgit v1.2.3-54-g00ecf