Update Dockerfile

Dockerfile

@@ -25,7 +25,7 @@ ARG INSTALL_CHANNEL=pytorch
 # Automatically set by buildx
 ARG TARGETPLATFORM
 
-#USER root
+USER root
 
 # Update basic dependencies we'll be using.
 #RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
@@ -40,7 +40,9 @@ ARG TARGETPLATFORM
 
 RUN git clone https://github.com/lee-ny/teaching_arithmetic.git /teaching_arithmetic 
 # && cd teaching_arithmetic && pip install -e .
-#RUN chmod +x /teaching_arithmetic
+RUN chmod +x /teaching_arithmetic
+USER 1000
+RUN teaching_arithmetic && pip install -e .
 # Install conda
 # translating Docker's TARGETPLATFORM into mamba arches
 #RUN case ${TARGETPLATFORM} in \