Skip to content

Commit bdffe07

Browse files
committed
fix: pin warp-lang<1.13.0 for evo2_megatron CI
subquadratic-ops-torch 0.2.0 uses wp.context.Module() which was removed in warp-lang 1.13.0 (shipped in pytorch:26.04-py3). Pin to <1.13.0 until a compatible subquadratic-ops release is available. Signed-off-by: svc-bionemo <267129667+svc-bionemo@users.noreply.github.com>
1 parent 32c53fe commit bdffe07

1 file changed

Lines changed: 7 additions & 4 deletions

File tree

bionemo-recipes/recipes/evo2_megatron/.ci_build.sh

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,11 +10,14 @@ uv venv --system-site-packages
1010
# 2. Activate the environment
1111
source .venv/bin/activate
1212

13-
# 3. Install build requirements and pin transformer_engine
13+
# 3. Pin warp-lang<1.13.0 (subquadratic-ops-torch 0.2.0 uses wp.context removed in 1.13)
14+
uv pip install 'warp-lang<1.13.0'
15+
16+
# 4. Install build requirements and pin transformer_engine
1417
pip freeze | grep transformer_engine > pip-constraints.txt
1518
uv pip install -r build_requirements.txt --no-build-isolation
1619

17-
# 4. Pre-install local sub-packages if checked out by CI.
20+
# 5. Pre-install local sub-packages if checked out by CI.
1821
# pyproject.toml references bionemo-recipeutils and bionemo-core from git (main).
1922
# In CI, the workflow sparse-checks them out alongside this recipe so we can test
2023
# against the PR's changes. But if the sub-packages don't exist on main yet (e.g.
@@ -35,8 +38,8 @@ for pkg_dir in "$RECIPE_ROOT/../../../sub-packages/bionemo-recipeutils" "$RECIPE
3538
fi
3639
done
3740

38-
# 5. Install the recipe with all remaining dependencies
41+
# 6. Install the recipe with all remaining dependencies
3942
uv pip install -c pip-constraints.txt -e . --no-build-isolation
4043

41-
# 6. Restore original pyproject.toml (the edit was only needed for uv resolution)
44+
# 7. Restore original pyproject.toml (the edit was only needed for uv resolution)
4245
mv pyproject.toml.ci_bak pyproject.toml

0 commit comments

Comments
 (0)