Dockerfile

ARG MINICONDA_IMAGE_TAG=4.10.3

FROM continuumio/miniconda3:$MINICONDA_IMAGE_TAG AS base

WORKDIR /app/

# install poetry
COPY ./requirements.txt ./requirements.txt
RUN --mount=type=cache,target=/root/.cache \
    python3 -m pip install -r ./requirements.txt

# create new environment
# warning: for some reason conda can hang on "Executing transaction" for a couple of minutes
COPY environment.yaml ./environment.yaml
RUN --mount=type=cache,target=/opt/conda/pkgs \
    conda env create -f ./environment.yaml

# "activate" environment for all commands (note: ENTRYPOINT is separate from SHELL)
SHELL ["conda", "run", "--no-capture-output", "-n", "kilroy-face-discord", "/bin/bash", "-c"]

WORKDIR /app/kilroy_face_discord/

# add poetry files
COPY ./kilroy_face_discord/pyproject.toml ./kilroy_face_discord/poetry.lock ./

FROM base AS test

# install dependencies only (notice that no source code is present yet)
RUN --mount=type=cache,target=/root/.cache \
    poetry install --no-root --only main,test

# add source, tests and necessary files
COPY ./kilroy_face_discord/src/ ./src/
COPY ./kilroy_face_discord/tests/ ./tests/
COPY ./kilroy_face_discord/LICENSE ./kilroy_face_discord/README.md ./

# build wheel by poetry and install by pip (to force non-editable mode)
RUN poetry build -f wheel && \
    python -m pip install --no-deps --no-index --no-cache-dir --find-links=dist kilroy-face-discord

RUN --mount=type=cache,target=/etc/torch-tmp \
    TORCH_HOME=/etc/torch-tmp kilroy-face-discord-fetch-models && \
    mkdir -p /etc/torch && \
    cp -rup /etc/torch-tmp/* /etc/torch
ENV TORCH_HOME=/etc/torch

# add entrypoint
COPY ./entrypoint.sh ./entrypoint.sh

ENTRYPOINT ["./entrypoint.sh", "pytest"]
CMD []

FROM base AS production

# install dependencies only (notice that no source code is present yet)
RUN --mount=type=cache,target=/root/.cache \
    poetry install --no-root --only main

# add source and necessary files
COPY ./kilroy_face_discord/src/ ./src/
COPY ./kilroy_face_discord/LICENSE ./kilroy_face_discord/README.md ./

# build wheel by poetry and install by pip (to force non-editable mode)
RUN poetry build -f wheel && \
    python -m pip install --no-deps --no-index --no-cache-dir --find-links=dist kilroy-face-discord

RUN --mount=type=cache,target=/etc/torch-tmp \
    TORCH_HOME=/etc/torch-tmp kilroy-face-discord-fetch-models && \
    mkdir -p /etc/torch && \
    cp -rup /etc/torch-tmp/* /etc/torch
ENV TORCH_HOME=/etc/torch

# add entrypoint
COPY ./entrypoint.sh ./entrypoint.sh

ENTRYPOINT ["./entrypoint.sh", "kilroy-face-discord"]
CMD []