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}