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