From b93db13e4a2f9a3c1347dab589b53c2236b81681 Mon Sep 17 00:00:00 2001
From: Bas Nijholt <basnijholt@gmail.com>
Date: Mon, 11 Sep 2017 10:56:20 +0200
Subject: [PATCH] 0D: return loss_improvements in choose_points
---
adaptive/learner.py | 4 +---
1 file changed, 1 insertion(+), 3 deletions(-)
diff --git a/adaptive/learner.py b/adaptive/learner.py
index ea3defd1..f89143dc 100644
--- a/adaptive/learner.py
+++ b/adaptive/learner.py
@@ -131,9 +131,7 @@ class AverageLearner(BaseLearner):
def choose_points(self, n, add_data=True):
points = list(range(self.n_requested, self.n_requested + n))
-
- # XXX: We should return the actual loss improvements
- loss_improvements = [None] * n
+ loss_improvements = [self.loss()] * n
if add_data:
self.add_data(points, itertools.repeat(None))
return points, loss_improvements
--
GitLab