From e3c1ac2d8dcdd06333ede612505e487ef4a1721f Mon Sep 17 00:00:00 2001
From: Tim Mensinger <mensingertim@gmail.com>
Date: Sun, 21 Jan 2024 17:09:32 +0100
Subject: [PATCH] Use history.get_fvals method

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

diff --git a/src/tranquilo/history.py b/src/tranquilo/history.py
index ddef245..96841a5 100644
--- a/src/tranquilo/history.py
+++ b/src/tranquilo/history.py
@@ -185,10 +185,8 @@ def get_best(self):
                 which is a mean if there are multiple evaluations per x, and the index.
 
         """
-        average_fvals = {
-            x_index: np.mean(self.fvals[f_indices])
-            for x_index, f_indices in self.index_mapper.items()
-        }
+        fvals = self.get_fvals(np.arange(self.n_xs))
+        average_fvals = {key: np.mean(val) for key, val in fvals.items()}
         index = pd.Series(average_fvals).idxmin()
         return self.get_xs(index), average_fvals[index], index