FROM huggingface/transformers-pytorch-gpu AS requirements
WORKDIR /root
COPY requirements.txt /root/requirements.txt
COPY ./sia/ /root/sia/
RUN pip3 install -r requirements.txt

FROM requirements AS test
COPY ./tests/ /root/tests/
RUN mkdir -p /root/model
CMD python3 -m unittest discover tests

FROM requirements