Error running query: <class 'ModuleNotFoundError'> No module named 'dask'

I have python set up as a data source as follows:

I am then trying to import dask.bag as follows:

import re
import boto3
import pymongo
from pymongo import MongoClient
import dask.bag as db

I keep getting the error

Error running query: <class 'ModuleNotFoundError'> No module named 'dask'

I have installed dask in the worker docker container. Any ideas what i am missing?