Trying to use the DaskTaskRunner in Prefect 2.0 and getting this error. I am following the published documentation.

from prefect.task_runner import DaskTaskRunner
ModuleNotFoundError: No module named 'prefect.task_runner'

Is there an extra that needs installed?

sorry for that, try:

pip install prefect-dask

I think it’s

prefect.task_runners

?

and then in your code:

from prefect_dask.task_runners import DaskTaskRunner

From the Task Runners page....

from prefect import flow
from prefect.task_runner import DaskTaskRunner

@flow(task_runner=DaskTaskRunner(address="<http://my-dask-cluster>"))
def my_flow():
    pass

ahh Kevin is right; this is just a typo in the docs we might not have moved that yet to a collection Tim, what I showed is what will come in the next releases - a spoiler: Dask and Ray will move to separate collections to make a cleaner separation and prevent users from having to install dependencies they may not need (dask, ray, ...)

ok, so it

prefect-dask

an extra, or separate dependency? I use

poetry

to manage our stack

forget my message for now, this is just a typo

I think

from prefect.task_runners import DaskTaskRunner

will work but eventually you will need

prefect-dask

installed

Ok, thanks everybody