Fix int tests 2 (#675)

predibase · Nov 12, 2024 · 8f2cbf7 · 8f2cbf7
1 parent 9358d0b
commit 8f2cbf7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
@@ -124,11 +124,12 @@ jobs:
       #       docker tag ghcr.io/predibase/lorax:main "$tag"
       #     done <<< "$tags"
 
+
       - name: Run integration tests
         uses: ./.github/workflows/integration-tests
         if: ${{ github.ref == 'refs/heads/main' }}
         with:
-          test_image_tag: ${{ steps.meta.outputs.tags[0] }}
+          test_image_tag: ${{ steps.meta.outputs.tags }}
           use_local_image: true
           github_token: ${{ secrets.GHCR_PAT }}
           huggingface_token: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}

diff --git a/.github/workflows/integration-tests/action.yaml b/.github/workflows/integration-tests/action.yaml
@@ -57,7 +57,8 @@ runs:
       shell: bash
       run: |
         echo "HUGGING_FACE_HUB_TOKEN=${{ inputs.huggingface_token }}" >> $GITHUB_ENV
-        echo "TEST_IMAGE_TAG=${{ inputs.test_image_tag }}" >> $GITHUB_ENV
+        first_tag=$(echo "${{ inputs.test_image_tag }}" | head -n 1)
+        echo "TEST_IMAGE_TAG=$first_tag" >> $GITHUB_ENV
 
     - name: Run Embedding tests
       shell: bash