Skip to content

Commit 88c700e

Browse files
fix: handle megatron nvrx AssertionError and bump nvidia-resiliency-ext>=0.6.0
megatron-core raises AssertionError (not ImportError) at import time when nvidia-resiliency-ext<0.6.0 is installed. Adding AssertionError to the except clause in deploy_ray.py lets the existing HAVE_RAY=False fallback handle this gracefully, fixing test_deploy_ray.py collection errors in CI. Also bumps the override pin from >=0.3.0,<=0.6.0 to >=0.6.0. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com> Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com>
1 parent 6a547cc commit 88c700e

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

nemo_deploy/deploy_ray.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@
3030
from nemo_deploy.multimodal.megatron_multimodal_deployable_ray import MegatronMultimodalRayDeployable
3131

3232
HAVE_RAY = True
33-
except (ImportError, ModuleNotFoundError):
33+
except (ImportError, ModuleNotFoundError, AssertionError):
3434
from unittest.mock import MagicMock
3535

3636
ray = MagicMock()

0 commit comments

Comments
 (0)