fix: relax dask version requirement and use a better check for Task

fix: relax dask version requirement and use a better check for Task #2355