Hi @<1540867420321746944:profile|DespicableSeaturtle77> , I think you need to define it per step

I am defining a pipeline step as follows:
pipe.add_function_step(
name='Split Data',
function=step_one,
cache_executed_step=True,
)

from the docs I can only specify the docker image using docker kwarg