Update action.yml

neuralmagic · Apr 6, 2024 · 7a50226 · 7a50226
1 parent dd3a288
commit 7a50226
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/actions/nm-lm-eval-accuracy/action.yml b/.github/actions/nm-lm-eval-accuracy/action.yml
@@ -17,6 +17,7 @@ runs:
       source $(pyenv root)/versions/${{ inputs.python }}/envs/${VENV}/bin/activate
 
       pip3 install git+https://github.com/EleutherAI/lm-evaluation-harness.git@262f879a06aa5de869e5dd951d0ff2cf2f9ba380
+      pip3 install pytest
 
       SUCCESS=0
       python .github/scripts/test_lm_eval_sweep.py -s -v || SUCCESS=$?