How to use a dask cluster as a scheduler for dask.compute

I have a class that has something like the following context manager to create a dask client & cluster: