File tree 1 file changed +4
-2
lines changed
1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -163,7 +163,8 @@ while queue.max_priority() < target_loss:
163
163
if loss.n_neighbors > 0:
164
164
subdomains_to_update = set()
165
165
for d in new_subdomains:
166
- subdomains_to_update.update(domain.neighbors(d, loss.n_neighbors))
166
+ neighbors = domain.neighbors(d, loss.n_neighbors)
167
+ subdomains_to_update.update(neighbors)
167
168
subdomains_to_update -= set(new_subdomains)
168
169
for subdomain in subdomains_to_update:
169
170
queue.update(subdomain, priority=loss(domain, subdomain, data))
@@ -228,7 +229,8 @@ while executor.n_outstanding_points > 0:
228
229
if loss.n_neighbors > 0 :
229
230
subdomains_to_update = set()
230
231
for d in new_subdomains :
231
- subdomains_to_update.update(domain.neighbors(d, loss.n_neighbors))
232
+ neighbors = domain.neighbors(d, loss.n_neighbors)
233
+ subdomains_to_update.update(neighbors)
232
234
subdomains_to_update -= set(new_subdomains)
233
235
for subdomain in subdomains_to_update :
234
236
queue.update(subdomain, priority=scaled_loss(domain, subdomain, data))
You can’t perform that action at this time.
0 commit comments