Skip some UT for pytorch 2.6 RC for now

intel · Jan 21, 2025 · b5bb6c7 · b5bb6c7
1 parent fb22bc5
commit b5bb6c7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/.github/workflows/llm_unit_tests.yml b/.github/workflows/llm_unit_tests.yml
@@ -425,7 +425,7 @@ jobs:
           bash python/llm/test/run-llm-inference-tests-gpu.sh
 
       - name: Run LLM example tests
-        if: ${{ matrix.pytorch-version != '2.6' }} # TODO: remove this workaround
+        if: ${{ matrix.pytorch-version != '2.6' }} # TODO: Enable this test for pytorch 2.6
         shell: bash
         run: |
           python -m pip uninstall datasets -y
@@ -442,6 +442,7 @@ jobs:
           bash python/llm/test/run-llm-example-tests-gpu.sh
 
       - name: Get Langchain version
+        if: ${{ matrix.pytorch-version != '2.6' }} # TODO: Enable this test for pytorch 2.6
         shell: bash
         id: get_langchain_version
         run: |
@@ -451,13 +452,15 @@ jobs:
           echo "langchain_ver=$LANGCHAIN_REF" >> $GITHUB_OUTPUT
 
       - name: Checkout Langchain repo
+        if: ${{ matrix.pytorch-version != '2.6' }} # TODO: Enable this test for pytorch 2.6
         uses: actions/checkout@f43a0e5ff2bd294095638e18286ca9a3d1956744
         with:
           repository: "langchain-ai/langchain"
           ref: ${{ join(steps.get_langchain_version.outputs.*, '\n') }}
           path: langchain_upstream
 
       - name: Run LLM langchain GPU test
+        if: ${{ matrix.pytorch-version != '2.6' }} # TODO: Enable this test for pytorch 2.6
         shell: bash
         run: |
           pip install -U langchain==0.0.184
@@ -478,6 +481,7 @@ jobs:
           bash python/llm/test/run-langchain-upstream-tests.sh
 
       - name: Run LLM llamaindex GPU test
+        if: ${{ matrix.pytorch-version != '2.6' }} # TODO: Enable this test for pytorch 2.6
         shell: bash
         run: |
           pip install "llama-index-readers-file<0.2.0"