Pin pytest-subtests version for accelerate tests (#6842)

microsoft · Dec 9, 2024 · 08b907a · 08b907a
1 parent 9a41cca
commit 08b907a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/nv-accelerate-v100.yml b/.github/workflows/nv-accelerate-v100.yml
@@ -52,7 +52,7 @@ jobs:
           pip install .[testing]
           # force protobuf version due to issues
           pip install "protobuf<4.21.0"
-          # tmp fix: force newer datasets version
-          #pip install "datasets>=2.0.0"
+          # force pytest-subtests version due to issues
+          pip install pytest-subtests==0.13.1
           pip list
           pytest $PYTEST_OPTS --color=yes --durations=0 --verbose tests/deepspeed