Skip to content

Commit

Permalink
Revert bits of update_who_has
Browse files Browse the repository at this point in the history
  • Loading branch information
crusaderky committed May 25, 2022
1 parent 263bf73 commit df488f0
Showing 1 changed file with 3 additions and 6 deletions.
9 changes: 3 additions & 6 deletions distributed/worker.py
Original file line number Diff line number Diff line change
Expand Up @@ -2998,11 +2998,8 @@ def _ensure_communicating(self, *, stimulus_id: str) -> RecsInstrs:
if ts.state != "fetch" or ts.key in all_keys_to_gather:
continue

if not ts.who_has:
recommendations[ts] = "missing"
continue

if self.validate:
assert ts.who_has
assert self.address not in ts.who_has

workers = [
Expand Down Expand Up @@ -4217,8 +4214,8 @@ def validate_task_fetch(self, ts):
assert self.address not in ts.who_has
assert not ts.done
assert ts in self.data_needed
# Note: ts.who_has may be have been emptied by _update_who_has, but the task
# won't transition to missing until it reaches the top of the data_needed heap.
assert ts.who_has

for w in ts.who_has:
assert ts.key in self.has_what[w]
assert ts in self.data_needed_per_worker[w]
Expand Down

0 comments on commit df488f0

Please sign in to comment.