From 0a89da74ee98920a108a7a11dc6ffa9c87236251 Mon Sep 17 00:00:00 2001
From: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>
Date: Fri, 17 Apr 2026 12:37:34 +0530
Subject: [PATCH 1/2] Temporarily disable latest mcore until we fix its
 nvidia-resiliency-ext dependency

Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>
---
 tox.ini | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tox.ini b/tox.ini
index 6694f7349d..0cd99cc86f 100644
--- a/tox.ini
+++ b/tox.ini
@@ -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]

From 0e9984b25d59cfa5a54332c931c83eae28462607 Mon Sep 17 00:00:00 2001
From: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>
Date: Fri, 17 Apr 2026 13:14:27 +0530
Subject: [PATCH 2/2] Update pyproject.toml

Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>
---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 16da6d6dc6..2993759ec1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -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",