Update PYTHONPATH env var in docker for Levanter

NVIDIA · Oct 22, 2024 · 02bca12 · 02bca12
1 parent 9332281
commit 02bca12
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.github/container/Dockerfile.levanter b/.github/container/Dockerfile.levanter
@@ -24,6 +24,8 @@ echo "-e file://${SRC_PATH_LEVANTER}" >> /opt/pip-tools.d/requirements-levanter.
 echo "-e file://${SRC_PATH_HALIAX}" >> /opt/pip-tools.d/requirements-levanter.in
 EOF
 
+ENV PYTHONPATH="${PYTHONPATH}:/opt/levanter/src"
+
 ###############################################################################
 ## Add warning for the shared data cache mechanism
 ###############################################################################

diff --git a/.github/workflows/_ci.yaml b/.github/workflows/_ci.yaml
@@ -496,7 +496,7 @@ jobs:
         ${{ needs.build-levanter.outputs.DOCKER_TAG_FINAL }} \
         bash <<"EOF" |& tee test-levanter.log
           pip install flake8 pytest soundfile librosa
-          PYTHONPATH=/opt/levanter/tests:/opt/levanter/src:$PYTHONPATH pytest /opt/levanter/tests -m "not slow and not ray"
+          PYTHONPATH=/opt/levanter/tests:$PYTHONPATH pytest /opt/levanter/tests -m "not slow and not ray"
         EOF
       STATISTICS_SCRIPT: |
         summary_line=$(tail -n1 test-levanter.log)