# R3AL Quant SDK — RunPod GPU image
FROM runpod/pytorch:2.1.0-py3.10-cuda11.8.0-devel

WORKDIR /app

# System deps for ONNX Runtime GPU
RUN apt-get update && apt-get install -y --no-install-recommends \
    libgomp1 \
    && rm -rf /var/lib/apt/lists/*

COPY pyproject.toml README.md ./
COPY src/ ./src/

RUN pip install --no-cache-dir -e ".[all]"

COPY handler.py ./
COPY scripts/handler_tests/ ./scripts/handler_tests/
COPY examples/handler/ ./examples/handler/

ENV PYTHONUNBUFFERED=1
CMD ["python", "-u", "handler.py"]
