diff --git a/.github/workflows/llm_unit_tests.yml b/.github/workflows/llm_unit_tests.yml
index 381246434eb..5461773f41a 100644
--- 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,6 +452,7 @@ 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"
@@ -458,6 +460,7 @@ jobs:
           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"