diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index 1ec456416..d8d6f73ea 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -84,8 +84,9 @@ jobs: uv pip show --system torch uv pip install --system wheel uv pip install --system --no-build-isolation -e ".[dev]" - python scripts/unsloth_install.py | sh - python scripts/cutcrossentropy_install.py | sh + set -o pipefail + python scripts/unsloth_install.py | bash + python scripts/cutcrossentropy_install.py | bash - name: Make sure PyTorch version wasn't clobbered run: | @@ -154,10 +155,12 @@ jobs: run: | uv pip show --system torch uv pip install --system wheel - uv pip install --system --no-build-isolation -e . + uv pip install --system build + python -m build --sdist + uv pip install --system dist/*.tar.gz python scripts/unsloth_install.py | sh python scripts/cutcrossentropy_install.py | sh - uv pip install --system -e .[dev] + uv pip install --system ".[dev]" - name: Make sure PyTorch version wasn't clobbered run: | diff --git a/cicd/Dockerfile.jinja b/cicd/Dockerfile.jinja index 479c668c7..78e6912e3 100644 --- a/cicd/Dockerfile.jinja +++ b/cicd/Dockerfile.jinja @@ -1,5 +1,7 @@ FROM axolotlai/axolotl-base-uv:{{ BASE_TAG }} +SHELL ["/bin/bash", "-euxo", "pipefail", "-c"] + ENV TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 8.6 9.0+PTX" ENV AXOLOTL_EXTRAS="{{ AXOLOTL_EXTRAS }}" ENV AXOLOTL_ARGS="{{ AXOLOTL_ARGS }}"