diff --git a/distributed/active_memory_manager.py b/distributed/active_memory_manager.py index 1502423064c..7f3df5d9665 100644 --- a/distributed/active_memory_manager.py +++ b/distributed/active_memory_manager.py @@ -306,7 +306,7 @@ def run( .. code-block:: python - choice = yield "replicate", ts, None + choice = (yield "replicate", ts, None) ``choice`` is either a WorkerState or None; the latter is returned if the ActiveMemoryManager chose to disregard the request.