Skip to content

Commit 6673f01

Browse files
committed
1D: remove the quals lambda function
1 parent 835cff2 commit 6673f01

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

adaptive/learner/learner1D.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -286,11 +286,11 @@ def xs(x, n):
286286
step = (x[1] - x[0]) / n
287287
return [x[0] + step * i for i in range(1, n)]
288288

289-
qual = lambda loss, x_range: -loss if loss != np.inf else -(x_range[1] - x_range[0]) / self._scale[0]
290-
291289
# Calculate how many points belong to each interval.
292-
quals = [(qual(loss, x_range), x_range, 1) for (x_range, loss) in
293-
self.losses_combined.items()]
290+
x_scale = self._scale[0]
291+
qualities = [(-loss if not math.isinf(loss) else (x0 - x1) / x_scale, (x0, x1))
292+
for ((x0, x1), loss in self.losses_combined.items()]
293+
quals = [(loss, x_range, 1) for (x_range, loss) in qualities]
294294

295295
heapq.heapify(quals)
296296

0 commit comments

Comments
 (0)