You could adapt how our Spark set up works https://github.com/kedro-org/kedro-starters/blob/main/pyspark/%7B%7B%20cookiecutter.repo_name%20%7D%7D/src/%7B%7B%20cookiecutter.python_package%20%7D%7D/context.py In fact @deepyaman expanded on this in these docs recently https://kedro.readthedocs.io/en/latest/deployment/dask.html