Try to fix llamaindex ut again (#12061)

This commit is contained in:
Yuwen Hu 2024-09-11 12:11:04 +08:00 committed by GitHub
parent 7e1e51d91a
commit c94032f97e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -261,9 +261,11 @@ jobs:
- name: Run LLM llamaindex test
shell: bash
run: |
pip install llama-index-readers-file llama-index-vector-stores-postgres llama-index-embeddings-huggingface
pip install "llama-index-readers-file<0.2.0"
pip install "llama-index-vector-stores-postgres<0.2.0"
pip install "llama-index-embeddings-huggingface<0.3.0"
pip install transformers==4.36.2
pip install "pydantic<2.9.1,>=2.0.0"
pip install "pydantic>=2.0.0"
bash python/llm/test/run-llm-llamaindex-tests.sh
- name: Run sentence-transformers uninstallation
if: ${{ always() }}
@ -471,7 +473,9 @@ jobs:
- name: Run LLM llamaindex GPU test
shell: bash
run: |
pip install llama-index-readers-file llama-index-vector-stores-postgres llama-index-embeddings-huggingface
pip install "llama-index-readers-file<0.2.0"
pip install "llama-index-vector-stores-postgres<0.2.0"
pip install "llama-index-embeddings-huggingface<0.3.0"
# Specific oneapi position on arc ut test machines
if [[ '${{ matrix.pytorch-version }}' == '2.1' ]]; then
pip install --pre --upgrade ipex-llm[xpu] --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/
@ -485,7 +489,7 @@ jobs:
fi
fi
pip install transformers==4.36.2
pip install "pydantic<2.9.1,>=2.0.0"
pip install "pydantic>=2.0.0"
bash python/llm/test/run-llm-llamaindex-tests-gpu.sh
- name: Run sentence-transformers uninstallation
if: ${{ always() }}