From 6ed2ac7c8ddc56d30797b0d8e509e72d1a39cd3e Mon Sep 17 00:00:00 2001 From: Fred Heinecke Date: Mon, 3 Aug 2026 14:34:30 -0500 Subject: [PATCH] Pin nltk to work around breaking change in tests Signed-off-by: Fred Heinecke --- qa/L0_jax_unittest/test.sh | 2 +- qa/L2_jax_unittest/test.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/qa/L0_jax_unittest/test.sh b/qa/L0_jax_unittest/test.sh index 5d833792cc..0be19bba67 100644 --- a/qa/L0_jax_unittest/test.sh +++ b/qa/L0_jax_unittest/test.sh @@ -21,7 +21,7 @@ FAILED_CASES="" export NVTE_JAX_TEST_TIMING=1 -pip3 install "nltk>=3.8.2" || error_exit "Failed to install nltk" +pip3 install "nltk>=3.8.2,<3.10.1" || error_exit "Failed to install nltk" pip3 install pytest==8.2.1 || error_exit "Failed to install pytest" : ${TE_PATH:=/opt/transformerengine} diff --git a/qa/L2_jax_unittest/test.sh b/qa/L2_jax_unittest/test.sh index f455ec0df3..dd11cd5d2d 100644 --- a/qa/L2_jax_unittest/test.sh +++ b/qa/L2_jax_unittest/test.sh @@ -21,7 +21,7 @@ FAILED_CASES="" export NVTE_JAX_TEST_TIMING=1 -pip3 install "nltk>=3.8.2" || error_exit "Failed to install nltk" +pip3 install "nltk>=3.8.2,<3.10.1" || error_exit "Failed to install nltk" pip3 install pytest==8.2.1 || error_exit "Failed to install pytest" : ${TE_PATH:=/opt/transformerengine}