Skip to content

Commit 4a62239

Browse files
committed
fix pytest --typeguard-packages=adaptive adaptive/tests/test_balancing_learner.py
1 parent a5b50bc commit 4a62239

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

adaptive/learner/balancing_learner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -215,7 +215,7 @@ def _ask_and_tell_based_on_npoints(
215215
selected = [] # tuples ((learner_index, point), loss_improvement)
216216
total_points = [l.npoints + len(l.pending_points) for l in self.learners]
217217
for _ in range(n):
218-
index = np.argmin(total_points)
218+
index = int(np.argmin(total_points))
219219
# Take the points from the cache
220220
if index not in self._ask_cache:
221221
self._ask_cache[index] = self.learners[index].ask(n=1)

0 commit comments

Comments
 (0)