We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a5b50bc commit 4a62239Copy full SHA for 4a62239
adaptive/learner/balancing_learner.py
@@ -215,7 +215,7 @@ def _ask_and_tell_based_on_npoints(
215
selected = [] # tuples ((learner_index, point), loss_improvement)
216
total_points = [l.npoints + len(l.pending_points) for l in self.learners]
217
for _ in range(n):
218
- index = np.argmin(total_points)
+ index = int(np.argmin(total_points))
219
# Take the points from the cache
220
if index not in self._ask_cache:
221
self._ask_cache[index] = self.learners[index].ask(n=1)
0 commit comments