FROM pytorch/pytorch:2.8.0-cuda12.8-cudnn9-runtime@sha256:417bd75df6365104c283ea4c1651fb3530d9eb5a4c2fafa51943cff2a94e6385
WORKDIR /tests
COPY tests/requirements-lock.txt /tmp/requirements-lock.txt
RUN python -m pip install --no-cache-dir --no-deps -r /tmp/requirements-lock.txt
COPY tests/public_data/ /tests/public_data/
COPY tests/artifact_validation.py tests/reward_policy.py tests/public_scoring.py tests/verifier.py tests/scoring.json tests/public_scoring.json tests/test.sh tests/intermediate.sh /tests/
COPY tests/reference/ /tests/reference/
COPY tests/adversarial/ /tests/adversarial/
RUN mkdir -p /output && PYTHONPATH=/tests python /tests/adversarial/artifact_validation_test.py
RUN PYTHONPATH=/tests python /tests/adversarial/reward_policy_cases.py
RUN PYTHONPATH=/tests python /tests/adversarial/seed_reconstruction_case.py
RUN PYTHONPATH=/tests python /tests/adversarial/public_scoring_cases.py
RUN chmod +x /tests/test.sh /tests/intermediate.sh /tests/verifier.py
ENV PYTHONUNBUFFERED=1 NVIDIA_VISIBLE_DEVICES=all NVIDIA_DRIVER_CAPABILITIES=compute,utility
CMD ["sleep", "infinity"]
