Skip to content

Commit f9d45aa

Browse files
committed
fix(evo2_megatron): pin warp-lang<1.13.0 in Dockerfile
subquadratic-ops is incompatible with warp-lang>=1.13.0. Pin warp-lang in the Dockerfile to match the .ci_build.sh fix from #1568. Signed-off-by: svc-bionemo <267129667+svc-bionemo@users.noreply.github.com>
1 parent bdffe07 commit f9d45aa

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

bionemo-recipes/recipes/evo2_megatron/Dockerfile

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,9 @@ ENV PATH="$VIRTUAL_ENV/bin:$PATH"
2121
# We create it one level up (/workspace/.venv) to keep it out of the source dir
2222
RUN uv venv --system-site-packages --seed $VIRTUAL_ENV
2323

24+
# FIXME: Pin warp-lang<1.13.0 until subquadratic-ops is compatible with warp 1.13+
25+
RUN uv pip install 'warp-lang<1.13.0'
26+
2427
# 4. Pin transformer_engine to the version pre-installed in the base image.
2528
# Without this constraint uv may try to upgrade or rebuild TE, which breaks the build.
2629
RUN pip freeze | grep transformer_engine > pip-constraints.txt

0 commit comments

Comments
 (0)