From a2d5786e19838788ad0d7305ec2bcfc4972d3e64 Mon Sep 17 00:00:00 2001
From: David Luevano Alvarado <55825613+luevano@users.noreply.github.com>
Date: Thu, 26 Mar 2020 20:22:42 -0700
Subject: Fix bob

---
 ml_exp/krr.py             | 5 ++---
 ml_exp/representations.py | 2 +-
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/ml_exp/krr.py b/ml_exp/krr.py
index 358816a39..68aef8828 100644
--- a/ml_exp/krr.py
+++ b/ml_exp/krr.py
@@ -91,7 +91,7 @@ def krr(descriptors,
         printc(f'\tTraining size: {training_size}', 'CYAN')
         printc(f'\tTest size: {test_size}', 'CYAN')
         printc(f'\tSigma: {sigma}', 'CYAN')
-        printc(f'\tKernel: {laplauss}', 'CYAN')
+        printc(f'\tKernel: {kernel}', 'CYAN')
         printc(f'\tUse tf: {use_tf}', 'CYAN')
 
     if use_tf:
@@ -310,8 +310,7 @@ def multi_krr(db_path='data',
                             sort=sort,
                             flatten=flatten)
         if 'BOB' in identifiers:
-            compound.gen_bob(size=size,
-                             sort=sort,
+            compound.gen_bob(sort=sort,
                              acount=acount)
 
     # Create a numpy array (or tensorflow tensor) for the descriptors.
diff --git a/ml_exp/representations.py b/ml_exp/representations.py
index 563b91c2f..31ab05169 100644
--- a/ml_exp/representations.py
+++ b/ml_exp/representations.py
@@ -356,7 +356,7 @@ Use non-eigenvalue representation.')
                 else:
                     bag = ''.join(sorted([atom_i, atom_j]))
 
-            bags[bag][1].append(cm[i, j])
+                bags[bag][1].append(cm[i, j])
 
     # Change to a numpy array and add padding.
     for bag in bags.keys():
-- 
cgit v1.2.3