FROM python:3.11

WORKDIR /opt
RUN apt-get update \
    && apt-get install swig -y
RUN pip install "litrl[demo, torchrl]~=0.0.16"

COPY ./src /opt/src
COPY ./run.py /opt/run.py

RUN mkdir -p /.cache/torchrl \
    && mkdir -p /.cache/huggingface \
    && chmod -R 777 /.cache/torchrl \
    && chmod -R 777 /.cache/huggingface

# Runtime args
CMD python run.py