Fix failing test

dask · Nov 15, 2019 · 86a66f9 · 86a66f9
1 parent 92441f0
commit 86a66f9
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/distributed/scheduler.py b/distributed/scheduler.py
@@ -2006,9 +2006,8 @@ def cancel_key(self, key, client, retries=5, force=False):
             return
         if ts is None or not ts.who_wants:  # no key yet, lets try again in a moment
             if retries:
-                self.loop.add_future(
-                    asyncio.sleep(0.2),
-                    lambda _: self.cancel_key(key, client, retries - 1),
+                self.loop.call_later(
+                    0.2, lambda _: self.cancel_key(key, client, retries - 1)
                 )
             return
         if force or ts.who_wants == {cs}:  # no one else wants this key