ARG BASE_DOCKER
# Dockerfile that updates the container with new code.
# SageMaker PyTorch image
FROM ${BASE_DOCKER}

# /opt/ml and all subdirectories are utilized by SageMaker, use the /code subdirectory to store your user code.
COPY . /opt/ml/code/

# RUN pip install -e /opt/ml/code/

# Prevent sagemaker from installing requirements again.
RUN rm /opt/ml/code/setup.py
RUN rm /opt/ml/code/requirements.txt

ENV SAGEMAKER_PROGRAM open_lm/main.py