chore: Install python dependencies with uv in workflows (#4086) #2
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name: integration-tests-and-build | |
on: | |
push: | |
branches: | |
- master | |
jobs: | |
build-lambda-docker-image: | |
if: github.repository == 'feast-dev/feast' | |
runs-on: ubuntu-latest | |
steps: | |
- uses: actions/checkout@v3 | |
- name: Set up QEMU | |
uses: docker/setup-qemu-action@v1 | |
- name: Set up Docker Buildx | |
uses: docker/setup-buildx-action@v2 | |
with: | |
install: true | |
- name: Set up AWS SDK | |
uses: aws-actions/configure-aws-credentials@v1 | |
with: | |
aws-access-key-id: ${{ secrets.AWS_ACCESS_KEY_ID }} | |
aws-secret-access-key: ${{ secrets.AWS_SECRET_ACCESS_KEY }} | |
aws-region: us-west-2 | |
- name: Login to Amazon ECR | |
id: login-ecr | |
uses: aws-actions/amazon-ecr-login@v1 | |
- name: Set ECR image tag | |
id: image-tag | |
run: echo "::set-output name=DOCKER_IMAGE_TAG::`git rev-parse HEAD`" | |
- name: Cache Public ECR Image | |
id: lambda_python_3_9 | |
uses: actions/cache@v2 | |
with: | |
path: ~/cache | |
key: lambda_python_3_9 | |
- name: Handle Cache Miss (pull public ECR image & save it to tar file) | |
if: steps.cache-primes.outputs.cache-hit != 'true' | |
run: | | |
mkdir -p ~/cache | |
docker pull public.ecr.aws/lambda/python:3.9 | |
docker save public.ecr.aws/lambda/python:3.9 -o ~/cache/lambda_python_3_9.tar | |
- name: Handle Cache Hit (load docker image from tar file) | |
if: steps.cache-primes.outputs.cache-hit == 'true' | |
run: | | |
docker load -i ~/cache/lambda_python_3_9.tar | |
- name: Build and push | |
env: | |
ECR_REGISTRY: ${{ steps.login-ecr.outputs.registry }} | |
ECR_REPOSITORY: feast-python-server | |
run: | | |
docker build \ | |
--file sdk/python/feast/infra/feature_servers/aws_lambda/Dockerfile \ | |
--tag $ECR_REGISTRY/$ECR_REPOSITORY:${{ steps.image-tag.outputs.DOCKER_IMAGE_TAG }} \ | |
--load \ | |
. | |
docker push $ECR_REGISTRY/$ECR_REPOSITORY:${{ steps.image-tag.outputs.DOCKER_IMAGE_TAG }} | |
outputs: | |
DOCKER_IMAGE_TAG: ${{ steps.image-tag.outputs.DOCKER_IMAGE_TAG }} | |
integration-test-python: | |
if: github.repository == 'feast-dev/feast' | |
needs: build-lambda-docker-image | |
runs-on: ${{ matrix.os }} | |
strategy: | |
fail-fast: false | |
matrix: | |
python-version: ["3.9", "3.10" ] | |
os: [ ubuntu-latest ] | |
env: | |
OS: ${{ matrix.os }} | |
PYTHON: ${{ matrix.python-version }} | |
services: | |
redis: | |
image: redis | |
ports: | |
- 6379:6379 | |
options: >- | |
--health-cmd "redis-cli ping" | |
--health-interval 10s | |
--health-timeout 5s | |
--health-retries 5 | |
steps: | |
- uses: actions/checkout@v5 | |
- name: Setup Python | |
id: setup-python | |
uses: actions/setup-python@v3 | |
with: | |
python-version: ${{ matrix.python-version }} | |
architecture: x64 | |
- name: Authenticate to Google Cloud | |
uses: 'google-github-actions/auth@v1' | |
with: | |
credentials_json: '${{ secrets.GCP_SA_KEY }}' | |
- name: Set up gcloud SDK | |
uses: google-github-actions/setup-gcloud@v1 | |
with: | |
project_id: ${{ secrets.GCP_PROJECT_ID }} | |
- name: Use gcloud CLI | |
run: gcloud info | |
- name: Set up AWS SDK | |
uses: aws-actions/configure-aws-credentials@v1 | |
with: | |
aws-access-key-id: ${{ secrets.AWS_ACCESS_KEY_ID }} | |
aws-secret-access-key: ${{ secrets.AWS_SECRET_ACCESS_KEY }} | |
aws-region: us-west-2 | |
- name: Use AWS CLI | |
run: aws sts get-caller-identity | |
- name: Upgrade pip version | |
run: | | |
pip install --upgrade "pip>=21.3.1,<23.2" | |
- name: Install uv | |
run: pip install uv | |
- name: Get uv cache dir | |
id: uv-cache | |
run: | | |
echo "::set-output name=dir::$(uv cache dir)" | |
- name: uv cache | |
uses: actions/cache@v4 | |
with: | |
path: ${{ steps.uv-cache.outputs.dir }} | |
key: ${{ runner.os }}-${{ steps.setup-python.outputs.python-version }}-uv-${{ hashFiles(format('**/py{0}-ci-requirements.txt', env.PYTHON)) }} | |
- name: Install dependencies | |
run: make install-python-ci-dependencies-uv | |
- name: Setup Redis Cluster | |
run: | | |
docker pull vishnunair/docker-redis-cluster:latest | |
docker run -d -p 6001:6379 -p 6002:6380 -p 6003:6381 -p 6004:6382 -p 6005:6383 -p 6006:6384 --name redis-cluster vishnunair/docker-redis-cluster | |
- name: Test python and go | |
env: | |
FEAST_SERVER_DOCKER_IMAGE_TAG: ${{ needs.build-lambda-docker-image.outputs.DOCKER_IMAGE_TAG }} | |
SNOWFLAKE_CI_DEPLOYMENT: ${{ secrets.SNOWFLAKE_CI_DEPLOYMENT }} | |
SNOWFLAKE_CI_USER: ${{ secrets.SNOWFLAKE_CI_USER }} | |
SNOWFLAKE_CI_PASSWORD: ${{ secrets.SNOWFLAKE_CI_PASSWORD }} | |
SNOWFLAKE_CI_ROLE: ${{ secrets.SNOWFLAKE_CI_ROLE }} | |
SNOWFLAKE_CI_WAREHOUSE: ${{ secrets.SNOWFLAKE_CI_WAREHOUSE }} | |
run: make test-python-integration | |
- name: Benchmark python | |
env: | |
FEAST_SERVER_DOCKER_IMAGE_TAG: ${{ needs.build-lambda-docker-image.outputs.DOCKER_IMAGE_TAG }} | |
SNOWFLAKE_CI_DEPLOYMENT: ${{ secrets.SNOWFLAKE_CI_DEPLOYMENT }} | |
SNOWFLAKE_CI_USER: ${{ secrets.SNOWFLAKE_CI_USER }} | |
SNOWFLAKE_CI_PASSWORD: ${{ secrets.SNOWFLAKE_CI_PASSWORD }} | |
SNOWFLAKE_CI_ROLE: ${{ secrets.SNOWFLAKE_CI_ROLE }} | |
SNOWFLAKE_CI_WAREHOUSE: ${{ secrets.SNOWFLAKE_CI_WAREHOUSE }} | |
run: pytest --verbose --color=yes sdk/python/tests --integration --benchmark --benchmark-autosave --benchmark-save-data --durations=5 | |
- name: Upload Benchmark Artifact to S3 | |
run: aws s3 cp --recursive .benchmarks s3://feast-ci-pytest-benchmarks | |
build-all-docker-images: | |
if: github.repository == 'feast-dev/feast' | |
runs-on: ubuntu-latest | |
strategy: | |
matrix: | |
component: [ feature-server-python-aws, feature-server-java, feature-transformation-server ] | |
env: | |
MAVEN_CACHE: gs://feast-templocation-kf-feast/.m2.2020-08-19.tar | |
REGISTRY: gcr.io/kf-feast | |
steps: | |
- uses: actions/checkout@v3 | |
- name: Set up QEMU | |
uses: docker/setup-qemu-action@v1 | |
- name: Set up Docker Buildx | |
uses: docker/setup-buildx-action@v2 | |
with: | |
install: true | |
- name: Login to DockerHub | |
uses: docker/login-action@v1 | |
with: | |
username: ${{ secrets.DOCKERHUB_USERNAME }} | |
password: ${{ secrets.DOCKERHUB_TOKEN }} | |
- name: Authenticate to Google Cloud | |
uses: 'google-github-actions/auth@v1' | |
with: | |
credentials_json: '${{ secrets.GCP_SA_KEY }}' | |
- name: Set up gcloud SDK | |
uses: google-github-actions/setup-gcloud@v1 | |
with: | |
project_id: ${{ secrets.GCP_PROJECT_ID }} | |
- name: Use gcloud CLI | |
run: gcloud info | |
- run: gcloud auth configure-docker --quiet | |
- name: Build image | |
run: | | |
make build-${{ matrix.component }}-docker REGISTRY=${REGISTRY} VERSION=${GITHUB_SHA} | |
- name: Push image | |
run: | | |
make push-${{ matrix.component }}-docker REGISTRY=${REGISTRY} VERSION=${GITHUB_SHA} | |
docker tag ${REGISTRY}/${{ matrix.component }}:${GITHUB_SHA} ${REGISTRY}/${{ matrix.component }}:develop | |
docker push ${REGISTRY}/${{ matrix.component }}:develop |