From c7235e34a8d8e9169f85311c867a3447cdd093ad Mon Sep 17 00:00:00 2001
From: Yuwen Hu <54161268+Oscilloscope98@users.noreply.github.com>
Date: Fri, 19 Apr 2024 10:59:00 +0800
Subject: [PATCH] Small update to ut (#10804)

---
 .github/workflows/llm_unit_tests.yml | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/.github/workflows/llm_unit_tests.yml b/.github/workflows/llm_unit_tests.yml
index c1823657..5eb5b55e 100644
--- a/.github/workflows/llm_unit_tests.yml
+++ b/.github/workflows/llm_unit_tests.yml
@@ -122,6 +122,9 @@ jobs:
           python -m pip install --upgrade setuptools==58.0.4
           python -m pip install --upgrade wheel
 
+          # May remove later
+          pip uninstall sentence-transformers -y || true
+
       - name: Download llm binary
         uses: ./.github/actions/llm/download-llm-binary
 
@@ -276,6 +279,9 @@ jobs:
           python -m pip install --upgrade wheel
           python -m pip install --upgrade notebook
 
+          # May remove later
+          pip uninstall sentence-transformers -y || true
+
       - name: Download llm binary
         uses: ./.github/actions/llm/download-llm-binary