FROM python:3.13-slim

RUN apt-get update && apt-get install -y openssh-client && rm -rf /var/lib/apt/lists/*

WORKDIR /app

COPY pyproject.toml ./
COPY sched/ ./sched/
COPY worker/ ./worker/
COPY jobs/ ./jobs/
COPY tests/ ./tests/

RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu

CMD ["python", "-m", "sched", "run", "--script", "jobs/mnist.py", "--compose", "--trace"]