Move PYTHONENV to clone step

awslabs · Dec 2, 2024 · 32e26b2 · 32e26b2
1 parent 760d452
commit 32e26b2
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/docker/local/Dockerfile.local b/docker/local/Dockerfile.local
@@ -53,6 +53,7 @@ ARG TRANSFORMERS_VERSION=4.28.1
 RUN cd /root; git clone --branch v${DGL_VERSION} --single-branch https://github.com/dmlc/dgl.git
 ENV DGL_HOME=/root/dgl
 ENV DGLBACKEND=pytorch
+ENV PYTHONPATH="/root/dgl/tools/:${PYTHONPATH}"
 
 FROM base as base-cpu
 
@@ -139,9 +140,6 @@ RUN mkdir -p ${SSHDIR} \
 
 EXPOSE ${SSH_PORT}
 
-ENV PYTHONPATH="/root/dgl/tools/:${PYTHONPATH}"
-
-
 # Copy GraphStorm source and add to PYTHONPATH
 RUN mkdir -p /graphstorm
 COPY code/python/graphstorm /graphstorm/python/graphstorm