Skip to content

Commit 835cff2

Browse files
author
Jorn
committed
fix tests�that fail for learner: if no data is provided it should sample
uniformly
1 parent 91f4d99 commit 835cff2

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

adaptive/learner/learner1D.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -286,8 +286,10 @@ 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+
289291
# Calculate how many points belong to each interval.
290-
quals = [(-loss, x_range, 1) for (x_range, loss) in
292+
quals = [(qual(loss, x_range), x_range, 1) for (x_range, loss) in
291293
self.losses_combined.items()]
292294

293295
heapq.heapify(quals)

0 commit comments

Comments
 (0)