调用 dask scheduler story 函数时出错

Question

So I am following this example from dask documentation almost verbatim https://distributed.dask.org/en/stable/logging.html#task-transition-logs and but could not get it to work. Below is my code:

import dask
import time
import random
from dask_jobqueue import SLURMCluster
from distributed import Client

def dummy(x):
   time.sleep(5)
   return x

cluster = SLURMCluster(...) # you will need to put your queue name, cores, mem, etc...
client = Client(cluster)
f = client.submit(dummy, 2)
# we can sleep for a while for it to be finished

client.scheduler.story(f.key)
# client.scheduler.story(f) yields the same error

and I get this error:

TypeError: send_recv_from_rpc() takes 0 positional arguments but 1 was given on the line client.scheduler.story(f.key) .

Answer 1

You are very close, to access the scheduler use the client.cluster.scheduler , so the relevant code would look like this:

print(client.cluster.scheduler.story(f.key))
# print the story