Dockerfile.training

FROM nvcr.io/nvidia/pytorch:24.08-py3@sha256:3afe094482e641337edfab878c87ca9c14ab3572b2ad048ac2554dd56d7cc024

WORKDIR /workspace/opensec-env

COPY pyproject.toml .
COPY server/ server/
COPY client/ client/
COPY sim/ sim/
COPY oracle/ oracle/
COPY training/ training/
COPY schemas/ schemas/
COPY data/ data/
COPY configs/ configs/
COPY scripts/ scripts/
COPY openenv.yaml .

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

ENV PYTHONPATH=/workspace/opensec-env
ENV OPENSEC_REPLAY_CACHE_PATH=/workspace/opensec-env/data/replay_cache.db

CMD ["python", "-c", "print('OpenSec-Env training container ready')"]