Upgrade pytest version (#891)

Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

qa/L0_jax_unittest/test.sh

@@ -4,7 +4,7 @@
 
 set -xe
 
-pip install pytest==7.2
+pip install pytest==8.2.1
 : ${TE_PATH:=/opt/transformerengine}
 
 pytest -c $TE_PATH/tests/jax/pytest.ini -v $TE_PATH/tests/jax -k 'not distributed'

qa/L0_paddle_unittest/test.sh

@@ -4,7 +4,7 @@
 
 set -xe
 
-pip install pytest==7.2
+pip install pytest==8.2.1
 : ${TE_PATH:=/opt/transformerengine}
 pytest -Wignore -v $TE_PATH/tests/paddle
 pytest -Wignore -v $TE_PATH/examples/paddle/mnist

qa/L0_pytorch_unittest/test.sh

@@ -6,7 +6,7 @@ set -e
 
 : ${TE_PATH:=/opt/transformerengine}
 
-pip install pytest==7.2 onnxruntime==1.13.1
+pip install pytest==8.2.1 onnxruntime==1.13.1
 pytest -v -s $TE_PATH/tests/pytorch/test_sanity.py
 pytest -v -s $TE_PATH/tests/pytorch/test_recipe.py
 pytest -v -s $TE_PATH/tests/pytorch/test_deferred_init.py

setup.py

@@ -72,7 +72,7 @@ def setup_requirements() -> Tuple[List[str], List[str], List[str]]:
         "importlib-metadata>=1.0; python_version<'3.8'",
         "packaging",
     ]
-    test_reqs: List[str] = ["pytest"]
+    test_reqs: List[str] = ["pytest>=8.2.1"]
 
     # Requirements that may be installed outside of Python
     if not found_cmake():