diff --git a/adaptive/learner.py b/adaptive/learner.py index ea3defd1f33d87f99792f21ee7b19fd88e3e32cf..f89143dca559842bccb389342144d4a097fa49ba 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