refine where callback function for dask future is defined

[billlyzhaoyh]

we can't submit callback after a scheduler is closed so this is an attempt to fix that