# Use Python 3.9 as the base image
FROM python:3.9-slim

# Set environment variables to make Python outputs work correctly in Docker
ENV PYTHONUNBUFFERED=1 \
    PYTHONDONTWRITEBYTECODE=1 \
    PIP_NO_CACHE_DIR=off \
    PIP_DISABLE_PIP_VERSION_CHECK=on \
    PIP_DEFAULT_TIMEOUT=100

# Install Git, as it's often necessary for installing dependencies
RUN apt-get update && apt-get install -y --no-install-recommends git && rm -rf /var/lib/apt/lists/*

# Create a virtual environment to isolate our package dependencies locally
RUN python3 -m venv /venv
ENV PATH="/venv/bin:$PATH"

RUN pip install -U pip && pip install git+https://github.com/zenml-io/zenml.git@bugfix/whylogs-extras#20138b5a8c8e01ba74e3d8f3ba6f6a8ce4efb36b#egg=zenml[server,templates,terraform,secrets-aws,secrets-gcp,secrets-azure,secrets-hashicorp,s3fs,gcsfs,adlfs,dev,mlstacks]

RUN zenml integration export-requirements \
    --output-file integration-requirements.txt \
    whylogs scipy pytorch azure airflow aws deepchecks

RUN pip install uv
RUN uv venv

ENV PATH="/venv/bin:$PATH"

RUN uv pip install -r integration-requirements.txt

# Set the default command for the container to keep it running.
CMD ["tail", "-f", "/dev/null"]

 => ERROR [8/8] RUN uv pip install -r integration-requirements.txt                                                            314.9s
------
 > [8/8] RUN uv pip install -r integration-requirements.txt:
13.74 Resolved 274 packages in 13.14s
314.2 error: Failed to download distributions
314.2   Caused by: Failed to fetch wheel: torch==2.2.1
314.2   Caused by: Failed to extract source distribution
314.2   Caused by: request or response body error: operation timed out
314.2   Caused by: operation timed out
------
Dockerfile_test:59
--------------------
  57 |     ENV PATH="/venv/bin:$PATH"
  58 |
  59 | >>> RUN uv pip install -r integration-requirements.txt
  60 |
  61 |     # Set the default command for the container to keep it running.
--------------------
ERROR: failed to solve: process "/bin/sh -c uv pip install -r integration-requirements.txt" did not complete successfully: exit code: 2