install pytest

HabanaAI · Aug 14, 2024 · e2dd0fe · e2dd0fe
1 parent a9e3dfa
commit e2dd0fe
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/cpu-test.yml b/.github/workflows/cpu-test.yml
@@ -29,6 +29,7 @@ jobs:
         pip install torch --extra-index-url https://download.pytorch.org/whl/cpu
         pip install -r requirements-hpu.txt
         VLLM_TARGET_DEVICE=hpu python setup.py develop
+        pip install pytest
     - name: cpu-test
       run: |
         VLLM_SKIP_WARMUP=true VLLM_PROMPT_SEQ_BUCKET_MAX=128 python examples/offline_inference_fakehpu.py