The only way I managed to do it is to add the function to

runner.py

and overwrite the file during containerization - not very elegant!