From 02bca12623c833c3e22d9ec2f1605f4b3ae138fc Mon Sep 17 00:00:00 2001 From: Vladislav Kozlov Date: Mon, 21 Oct 2024 17:15:53 -0700 Subject: [PATCH] Update PYTHONPATH env var in docker for Levanter --- .github/container/Dockerfile.levanter | 2 ++ .github/workflows/_ci.yaml | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/.github/container/Dockerfile.levanter b/.github/container/Dockerfile.levanter index 90b31ec04..3b4d71b6a 100644 --- 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 index 9567c23e7..bfa88360f 100644 --- 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)