Move PYTHONENV to clone step

awslabs · Feb 15, 2025 · 4a85dbb · 4a85dbb
1 parent 28960d4
commit 4a85dbb
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/docker/local/Dockerfile.local b/docker/local/Dockerfile.local
@@ -54,6 +54,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
 
@@ -142,9 +143,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