Pin NLTK version to fix JAX ci (NVIDIA#1096)

Signed-off-by: Kirthi Shankar Sivamani <[email protected]> Co-authored-by: Phuong Nguyen <[email protected]>
phu0ngng · Aug 13, 2024 · b484038 · b484038
1 parent def4d1c
commit b484038
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/qa/L0_jax_unittest/test.sh b/qa/L0_jax_unittest/test.sh
@@ -4,6 +4,7 @@
 
 set -xe
 
+pip install nltk==3.8.1
 pip install pytest==8.2.1
 : ${TE_PATH:=/opt/transformerengine}