Temporarily disable latest pulp and mcore until we fix its nvidia-resiliency-ext dependency (#1285)

- `megatron-core==0.17.0` released yesterday which requires nightly
version of `nvidia-resiliency-ext` for an import. Pre-installed version
in DLFW Pytorch container is `nvidia-resiliency-ext==0.5.0`
  - Temporarily pin `mcore<0.17.0` to unblock PR from merging. 
- Pin `pulp<4.0` as it has some breaking changes and release imminent

Correct fix is to just use `nemo:26.04` container instead of PyTorch
container for megatron-based tests since it always has correct
combination of all packages needed for the megatron ecosystem - Done in
#1286

---------

Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>
This commit is contained in:
Keval Morabia
2026-04-17 16:31:19 +05:30
committed by GitHub
parent 7e82a5cb03
commit 4e33368dbe
2 changed files with 3 additions and 2 deletions
+1 -1
View File
@@ -43,7 +43,7 @@ dependencies = [
# modelopt.torch
"PyYAML>=6.0",
"omegaconf>=2.3.0",
"pulp",
"pulp<4.0", # breaking changes in upcoming 4.0 release
"pydantic>=2.0",
"regex",
"rich",
+2 -1
View File
@@ -82,7 +82,8 @@ commands =
[testenv:cuda13-gpu-megatron]
commands_pre =
# Install deps here so that it gets installed even in --current-env
pip install -U megatron-core
# Temporarily disable latest mcore until we fix its nvidia-resiliency-ext dependency
pip install 'megatron-core<0.17.0'
pip install --no-build-isolation git+https://github.com/state-spaces/mamba.git
pip install --no-build-isolation git+https://github.com/Dao-AILab/causal-conv1d.git
pip install -e .[hf,dev-test]