Flip client cases

dask · Nov 25, 2020 · 3b8741f · 3b8741f
1 parent c5acd1c
commit 3b8741f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/distributed/scheduler.py b/distributed/scheduler.py
@@ -2590,13 +2590,13 @@ def report(self, msg, ts=None, client=None):
         if ts is None:
             # Notify all clients
             client_keys = list(self.client_comms)
-        elif client is not None:
+        elif client is None:
+            # Notify clients interested in key
+            client_keys = [c.client_key for c in ts.who_wants]
+        else:
             # Notify clients interested in key (including `client`)
             client_keys = [c.client_key for c in ts.who_wants if c.client_key != client]
             client_keys.append(client)
-        else:
-            # Notify clients interested in key
-            client_keys = [c.client_key for c in ts.who_wants]
 
         for k in client_keys:
             try: