# HuggingFace Space (docker-sdk) — for transient training runs.
# Push this to a "trainer" Space; it trains the bandit, pushes models, sleeps.
FROM python:3.11-slim

WORKDIR /app
COPY pyproject.toml setup.py requirements.txt ./
COPY dash_jsp ./dash_jsp
COPY hf ./hf
COPY scripts ./scripts

RUN pip install --no-cache-dir --upgrade pip \
 && pip install --no-cache-dir -e .[hf] \
 && pip install --no-cache-dir huggingface_hub

ENV PYTHONUNBUFFERED=1

CMD bash -lc " \
    python scripts/download_benchmarks.py --family taillard lawrence && \
    python hf/train_on_hf.py --benchmarks taillard lawrence --epochs 3 --output-dir models && \
    python hf/push_models.py --models-dir models && \
    sleep infinity \
"