Skip to content

Load all models including targz #4178

Load all models including targz

Load all models including targz #4178

Workflow file for this run

name: Docker CI
on:
push:
branches: [ "master" ]
pull_request:
branches: [ "master" ]
merge_group:
jobs:
test-build-and-container:
runs-on: ubuntu-latest
strategy:
fail-fast: false
matrix:
python-version: ["3.8", "3.9", "3.10"]
steps:
- uses: actions/checkout@v3
with:
submodules: recursive
- name: Test build_image.sh script with custom tagging and gpu flag
working-directory: docker
run: ./test_build_image_tagging.sh ${{ matrix.python-version }}
- name: Determine the branch name
id: branch-name
run: |
if [[ "${GITHUB_REF}" == refs/heads/* ]]; then
GITHUB_BRANCH=${GITHUB_REF#refs/heads/}
elif [[ "${GITHUB_REF}" == refs/pull/*/merge ]]; then
GITHUB_BRANCH=${GITHUB_HEAD_REF}
fi
echo "GITHUB_BRANCH=${GITHUB_BRANCH}" >> $GITHUB_OUTPUT
- name: Build Image for container test
id: image_build
working-directory: docker
run: |
IMAGE_TAG=test-image-${{ matrix.python-version }}
REPO_URL="${{ github.event.pull_request.head.repo.clone_url }}"
if [[ -z "${REPO_URL}" ]]; then
REPO_URL="https://github.com/pytorch/serve.git"
fi
./build_image.sh -py "${{ matrix.python-version }}" -t "${IMAGE_TAG}" -b "${{ steps.branch-name.outputs.GITHUB_BRANCH }}" -repo ${REPO_URL} -s
echo "IMAGE_TAG=${IMAGE_TAG}" >> $GITHUB_OUTPUT
- name: Container Healthcheck
working-directory: docker
run: ./test_container_health.sh ${{ steps.image_build.outputs.IMAGE_TAG }}
- name: Check Python version in container
working-directory: docker
run: ./test_container_python_version.sh ${{ steps.image_build.outputs.IMAGE_TAG }} ${{ matrix.python-version }}
- name: Test model running in container with sample image data
working-directory: docker
run: |
./test_container_model_prediction.sh ${{ steps.image_build.outputs.IMAGE_TAG }}