Update JAX integration proposal #752
Workflow file for this run
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: Python Test | |
on: | |
- push | |
- pull_request | |
jobs: | |
python-test: | |
name: Test | |
runs-on: ubuntu-latest | |
strategy: | |
fail-fast: false | |
matrix: | |
# TODO (tenzen-y): Once we resolve this compatibility issue, we will test against all Python versions. | |
# REF: https://github.com/kubeflow/training-operator/issues/2096 | |
# python-version: ["3.8", "3.9", "3.10", "3.11"] | |
python-version: ["3.10", "3.11"] | |
steps: | |
- name: Check out code | |
uses: actions/checkout@v4 | |
- name: Setup Python | |
uses: actions/setup-python@v5 | |
with: | |
python-version: ${{ matrix.python-version }} | |
# TODO (andreyvelich): We need to replace this action with script to do | |
# linting and formatting for Training Operator SDK. | |
- name: Check Python code with Black | |
uses: psf/black@stable | |
with: | |
version: 24.2.0 | |
options: --check --exclude '/*kubeflow_org_v1*|__init__.py|api_client.py|configuration.py|exceptions.py|rest.py' | |
src: sdk/ | |
- name: Install dependencies | |
run: | | |
pip install pytest python-dateutil urllib3 kubernetes | |
pip install -U './sdk/python[huggingface]' | |
- name: Run unit test for training sdk | |
run: pytest ./sdk/python/kubeflow/training/api/training_client_test.py |