image builder: give flytekit user ownership of /root after venv is created (#3298)

Signed-off-by: Niels Bantilan <[email protected]>

Author: cosmicBboy

flytekit/image_spec/default_builder.py

@@ -108,9 +108,6 @@
     [ -f /etc/ssl/certs/ca-certificates.crt ] || \
     mkdir -p /etc/ssl/certs/ && cp /tmp/ca-certificates.crt /etc/ssl/certs/ca-certificates.crt
 
-RUN id -u flytekit || useradd --create-home --shell /bin/bash flytekit
-RUN chown flytekit /root && chown -R flytekit /home
-
 $INSTALL_PYTHON_TEMPLATE
 
 # Configure user space
@@ -137,6 +134,9 @@
 RUN --mount=type=cache,sharing=locked,mode=0777,target=/root/.cache/uv,id=uv \
     --mount=from=uv,source=/uv,target=/usr/bin/uv $RUN_COMMANDS
 
+RUN id -u flytekit || useradd --create-home --shell /bin/bash flytekit
+RUN chown -R flytekit /root && chown -R flytekit /home
+
 WORKDIR /root
 SHELL ["/bin/bash", "-c"]