Add batch number to TraceMe annotations in input preprocessing. #2422
Workflow file for this run
This file contains hidden or 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: Build and test | |
| on: | |
| # Only run workflow on pushes to main (includes PR merge), and on | |
| # opened pull-requests. | |
| push: | |
| branches: | |
| - main | |
| pull_request: | |
| permissions: | |
| contents: read # Lets the job check out your code | |
| actions: write # Lets the job write (save) the cache | |
| jobs: | |
| build-and-test-cpu: | |
| runs-on: linux-x86-n4-16 | |
| container: us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build:latest | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - name: Set up Python 3.10 | |
| uses: actions/setup-python@v5 | |
| with: | |
| python-version: '3.10' | |
| - name: Display Python version | |
| run: python -c "import sys; print(sys.version)" | |
| - name: Install dependencies | |
| run: | | |
| python -m pip install --upgrade --root-user-action=ignore pip setuptools wheel | |
| bash tools/install_bazelisk.sh | |
| # --- FOR PULL REQUESTS --- | |
| # Restore the cache from the main branch's base commit. | |
| - if: github.event_name == 'pull_request' | |
| name: Mount bazel cache (pull-request) | |
| uses: actions/cache/restore@v4 | |
| with: | |
| path: "/__w/.cache/bazel" | |
| key: bazel-py3.10-${{ github.base_ref }}-${{ github.event.pull_request.base.sha }} | |
| restore-keys: | | |
| bazel-py3.10-${{ github.base_ref }} | |
| bazel-py3.10- | |
| bazel- | |
| # --- FOR MAIN PUSHES AND MERGES --- | |
| # Restore the cache from the previous commit. | |
| - if: github.event_name != 'pull_request' | |
| name: Mount bazel cache (main) | |
| uses: actions/cache/restore@v4 | |
| with: | |
| path: "/__w/.cache/bazel" | |
| # Try to find the cache from the exact commit *before* this push | |
| key: bazel-py3.10-${{ github.ref_name }}-${{ github.event.before }} | |
| # Fall back to the most recent cache for the main branch | |
| restore-keys: | | |
| bazel-py3.10-${{ github.ref_name }} | |
| bazel-py3.10- | |
| bazel- | |
| - name: Build all targets | |
| run: | | |
| export HERMETIC_PYTHON_VERSION=3.10 | |
| bazel build --config=public_cache //... --build_tag_filters=-oss_exclude | |
| - name: Build wheel | |
| run: | | |
| export HERMETIC_PYTHON_VERSION=3.10 | |
| bazel run --config=public_cache //tools:build_wheel | |
| - name: Run CPU tests | |
| run: | | |
| export HERMETIC_PYTHON_VERSION=3.10 | |
| bazel test --config=public_cache --build_tests_only --test_tag_filters=-requires-tpu --test_output=errors --keep_going //... | |
| - name: Check disk space | |
| run: | | |
| df -h --total | |
| du -sh /__w/.cache/bazel | |
| # --- SAVE NEW CACHE --- | |
| - name: Save bazel cache (main) | |
| if: github.event_name != 'pull_request' | |
| uses: actions/cache/save@v4 | |
| with: | |
| path: "/__w/.cache/bazel" | |
| key: bazel-py3.10-${{ github.ref_name }}-${{ github.sha }} |