adding aws-shell to a docker image

What I'd like to do is add aws-shell to a docker image (easy using pip)

but then prewarm the command completion so it doesn't do that upon first run rather it is part of the build process.

how can I do this?