fix: try to fix imagespec error

-e
Signed-off-by: machichima <nary12321@gmail.com>

Author: machichima

flytekit/image_spec/default_builder.py

@@ -29,8 +29,7 @@
     --mount=type=bind,target=uv.lock,src=uv.lock \
     --mount=type=bind,target=pyproject.toml,src=pyproject.toml \
     $PIP_SECRET_MOUNT \
-    uv sync $PIP_INSTALL_ARGS && \
-    chown -R flytekit /root/.venv
+    uv sync $PIP_INSTALL_ARGS
 WORKDIR /
 
 # Update PATH and UV_PYTHON to point to the venv created by uv sync
@@ -55,12 +54,12 @@
     --mount=type=bind,target=poetry.lock,src=poetry.lock \
     --mount=type=bind,target=pyproject.toml,src=pyproject.toml \
     $PIP_SECRET_MOUNT \
-    poetry install $PIP_INSTALL_ARGS && \
-    chown -R flytekit /root/.venv
+    poetry install $PIP_INSTALL_ARGS
+
 WORKDIR /
 
 # Update PATH and UV_PYTHON to point to venv
-ENV PATH="/root/.venv/bin:$$PATH"  \
+ENV PATH="/root/.venv/bin:$$PATH" \
     UV_PYTHON=/root/.venv/bin/python
 """
 )
@@ -82,7 +81,6 @@
     $APT_PACKAGES
 """)
 
-# make sure that micromamba python installation is owned by flytekit user
 MICROMAMBA_INSTALL_COMMAND_TEMPLATE = Template("""\
 RUN --mount=type=cache,sharing=locked,mode=0777,target=/opt/micromamba/pkgs,\
 id=micromamba \
@@ -93,8 +91,7 @@
     python=$PYTHON_VERSION $CONDA_PACKAGES \
     || micromamba install -n runtime --root-prefix /opt/micromamba \
     -c conda-forge $CONDA_CHANNELS \
-    python=$PYTHON_VERSION $CONDA_PACKAGES ) && \
-    chown -R flytekit /opt/micromamba
+    python=$PYTHON_VERSION $CONDA_PACKAGES )
 """)
 
 DOCKER_FILE_TEMPLATE = Template("""\
@@ -112,7 +109,7 @@
     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 -R flytekit /root && chown -R flytekit /home
+RUN chown flytekit /root && chown -R flytekit /home
 
 $INSTALL_PYTHON_TEMPLATE
 
@@ -125,7 +122,7 @@
     SSL_CERT_DIR=/etc/ssl/certs \
     $ENV
 
-$PYTHON_INSTALL_COMMAND
+$UV_PYTHON_INSTALL_COMMAND
 
 # Adds nvidia just in case it exists
 ENV PATH="$$PATH:/usr/local/nvidia/bin:/usr/local/cuda/bin" \
@@ -343,7 +340,7 @@ def create_docker_context(image_spec: ImageSpec, tmp_dir: Path):
         )
         raise ValueError(msg)
 
-    python_install_command = prepare_python_install(image_spec, tmp_dir)
+    uv_python_install_command = prepare_python_install(image_spec, tmp_dir)
     env_dict = {"PYTHONPATH": "/root"}
 
     if image_spec.env:
@@ -429,11 +426,11 @@ def create_docker_context(image_spec: ImageSpec, tmp_dir: Path):
     _f_img_id_env = f"{_F_IMG_ID}={image_spec.id}"
 
     docker_content = DOCKER_FILE_TEMPLATE.substitute(
+        UV_PYTHON_INSTALL_COMMAND=uv_python_install_command,
+        APT_INSTALL_COMMAND=apt_install_command,
         INSTALL_PYTHON_TEMPLATE=python_install_template.template,
         EXTRA_PATH=python_install_template.extra_path,
         PYTHON_EXEC=python_install_template.python_exec,
-        APT_INSTALL_COMMAND=apt_install_command,
-        PYTHON_INSTALL_COMMAND=python_install_command,
         BASE_IMAGE=base_image,
         ENV=env,
         _F_IMG_ID_ENV=_f_img_id_env,