Skip to content

Commit 599e9d3

Browse files
authored
Update interp recipe workflow to match the others (#1701)
### Description Do the same Dockerfile change to the interpretability workflow in CI that was recently done to other workflows. Split out of #1699 ### Type of changes <!-- Mark the relevant option with an [x] --> - [ ] Bug fix (non-breaking change which fixes an issue) - [ ] New feature (non-breaking change which adds functionality) - [ ] Refactor - [ ] Documentation update - [x] Other (please describe): CI update ### CI Pipeline Configuration Configure CI behavior by applying the relevant labels. By default, only basic unit tests are run. - [ciflow:skip](https://github.com/NVIDIA/bionemo-framework/blob/main/docs/docs/main/contributing/contributing.md#ciflow:skip) - Skip all CI tests for this PR - [ciflow:notebooks](https://github.com/NVIDIA/bionemo-framework/blob/main/docs/docs/main/contributing/contributing.md#ciflow:notebooks) - Run Jupyter notebooks execution tests - [ciflow:slow](https://github.com/NVIDIA/bionemo-framework/blob/main/docs/docs/main/contributing/contributing.md#ciflow:slow) - Run slow single GPU integration tests marked as @pytest.mark.slow - [ciflow:all](https://github.com/NVIDIA/bionemo-framework/blob/main/docs/docs/main/contributing/contributing.md#ciflow:all) - Run all tests, including unit tests, slow tests, notebooks, and every recipe/model directory. Unit tests marked as `@pytest.mark.multi_gpu` or `@pytest.mark.distributed` are not run in the PR pipeline. For more details, see [CONTRIBUTING](CONTRIBUTING.md) > [!NOTE] > By default, only basic unit tests are run. Add appropriate labels to enable an additional test coverage. #### Authorizing CI Runs We use [copy-pr-bot](https://docs.gha-runners.nvidia.com/apps/copy-pr-bot/#automation) to manage authorization of CI runs on NVIDIA's compute resources. - If a pull request is opened by a trusted user and contains only trusted changes, the pull request's code will automatically be copied to a pull-request/ prefixed branch in the source repository (e.g. pull-request/123) - If a pull request is opened by an untrusted user or contains untrusted changes, an NVIDIA org member must leave an `/ok to test` comment on the pull request to trigger CI. This will need to be done for each new commit. #### Triggering Code Rabbit AI Review To trigger a code review from code rabbit, comment on a pull request with one of these commands: - @coderabbitai review - Triggers a standard review - @coderabbitai full review - Triggers a comprehensive review See https://docs.coderabbit.ai/reference/review-commands for a full list of commands. ### Pre-submit Checklist <!--- Ensure all items are completed before submitting --> - [ ] I have tested these changes locally - [ ] I have updated the documentation accordingly - [ ] I have added/updated tests as needed - [ ] All existing tests pass successfully <!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit * **Chores** * Updated the continuous integration environment for interpretability recipe checks. * Improved dependency setup and caching for automated test runs. * Refreshed the underlying test environment to support more consistent and reliable validation. <!-- end of auto-generated comment: release notes by coderabbit.ai --> Signed-off-by: John St John <jstjohn@nvidia.com>
1 parent e8e7f59 commit 599e9d3

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

.github/workflows/unit-tests-interpretability-recipes.yaml

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,8 +84,8 @@ jobs:
8484
fi
8585
echo "Recipes to run: $DIRS"
8686
87-
# Attach the runner image (the dockerhub-cached squashed pytorch, like the repo-wide lane).
88-
IMG="svcbionemo023/bionemo-framework:pytorch26.04-py3-squashed"
87+
# Attach the same default runner image as the repo-wide recipe lane.
88+
IMG="nvcr.io/nvidia/pytorch:26.06-py3"
8989
MATRIX=$(echo "$DIRS" | jq -c --arg img "$IMG" \
9090
'map({dir: ., name: (. | split("/") | last), image: $img})')
9191
echo "dirs=$MATRIX" >> "$GITHUB_OUTPUT"
@@ -138,6 +138,11 @@ jobs:
138138
${{ runner.os }}-huggingface-interp-${{ matrix.recipe.name }}-
139139
${{ runner.os }}-huggingface-
140140
141+
- name: Setup uv
142+
uses: astral-sh/setup-uv@d0d8abe699bfb85fec6de9f7adb5ae17292296ff
143+
with:
144+
enable-cache: true
145+
141146
- name: Install dependencies
142147
working-directory: ${{ matrix.recipe.dir }}
143148
run: bash .ci_build.sh

0 commit comments

Comments
 (0)