Disable multiprocess-unit-tests job.

PiperOrigin-RevId: 871888917

Author: Orbax Authors

.github/workflows/build.yml

@@ -324,55 +324,55 @@ jobs:
             "context": "github-actions/build"
             }'
 
-  multiprocess-unit-tests:
-    name: "multiprocess-unit-tests (Python ${{ matrix.python-version }}, jax=${{ matrix.jax-version }})"
-    runs-on: linux-x86-ct5lp-4tpu-x2
-    container: us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build:infrastructure-public-image-2d2a7b1e6e2e
-    defaults:
-      run:
-        working-directory: checkpoint
-    strategy:
-      matrix:
-        python-version: ["3.11"]
-        jax-version: ["newest"]
-    steps:
-    - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683  # v4.2.2
-    - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@0b93645e9fea7318ecaed2b359559ac225c90a2b  # v5.3.0
-      with:
-        python-version: ${{ matrix.python-version }}
-    - name: Install dependencies
-      run: |
-        pip install -e .
-        pip install -e .[testing,gcs] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
-        pip uninstall -y orbax
-        pip install gcsfs
-        pip install portpicker pytest chex pyyaml
-        if [ "${{ matrix.jax-version }}" = "newest" ]; then
-          pip install -U "jax[tpu]" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
-        elif [ "${{ matrix.jax-version }}" = "nightly" ]; then
-          pip install -U --pre "jax[tpu]" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html --extra-index-url https://us-python.pkg.dev/ml-oss-artifacts-published/jax-public-nightly-artifacts-registry/simple/
-        else
-          pip install "jax[tpu]==${{ matrix.jax-version }}" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
-        fi
-    - name: Run multiprocess tests
-      env:
-        TEST_TMPDIR: gs://orbax-benchmarks/unit-tests/${{ github.run_id }}
-      run: |
-        python orbax/checkpoint/_src/testing/multiprocess_unittests/run_tests.py --filename=orbax/checkpoint/_src/testing/multiprocess_unittests/tagged_tests.yaml --processes=4
-    - name: Report success or failure as github status
-      if: always()
-      shell: bash
-      run: |
-        status="${{ job.status }}"
-        lowercase_status=$(echo $status | tr '[:upper:]' '[:lower:]')
-        curl -sS --request POST \
-        --url https://api.github.com/repos/${{ github.repository }}/statuses/${{ github.sha }} \
-        --header 'authorization: Bearer ${{ secrets.GITHUB_TOKEN }}' \
-        --header 'content-type: application/json' \
-        --data '{
-            "state": "'$lowercase_status'",
-            "target_url": "https://github.com/${{ github.repository }}/actions/runs/${{ github.run_id }}",
-            "description": "'$status'",
-            "context": "github-actions/build"
-            }'
+  # multiprocess-unit-tests:
+    # name: "multiprocess-unit-tests (Python ${{ matrix.python-version }}, jax=${{ matrix.jax-version }})"
+    # runs-on: linux-x86-ct5lp-4tpu-x2
+    # container: us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build:infrastructure-public-image-2d2a7b1e6e2e
+    # defaults:
+    #   run:
+    #     working-directory: checkpoint
+    # strategy:
+    #   matrix:
+    #     python-version: ["3.11"]
+    #     jax-version: ["newest"]
+    # steps:
+    # - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683  # v4.2.2
+    # - name: Set up Python ${{ matrix.python-version }}
+    #   uses: actions/setup-python@0b93645e9fea7318ecaed2b359559ac225c90a2b  # v5.3.0
+    #   with:
+    #     python-version: ${{ matrix.python-version }}
+    # - name: Install dependencies
+    #   run: |
+    #     pip install -e .
+    #     pip install -e .[testing,gcs] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
+    #     pip uninstall -y orbax
+    #     pip install gcsfs
+    #     pip install portpicker pytest chex pyyaml
+    #     if [ "${{ matrix.jax-version }}" = "newest" ]; then
+    #       pip install -U "jax[tpu]" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
+    #     elif [ "${{ matrix.jax-version }}" = "nightly" ]; then
+    #       pip install -U --pre "jax[tpu]" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html --extra-index-url https://us-python.pkg.dev/ml-oss-artifacts-published/jax-public-nightly-artifacts-registry/simple/
+    #     else
+    #       pip install "jax[tpu]==${{ matrix.jax-version }}" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
+    #     fi
+    # - name: Run multiprocess tests
+    #   env:
+    #     TEST_TMPDIR: gs://orbax-benchmarks/unit-tests/${{ github.run_id }}
+    #   run: |
+    #     python orbax/checkpoint/_src/testing/multiprocess_unittests/run_tests.py --filename=orbax/checkpoint/_src/testing/multiprocess_unittests/tagged_tests.yaml --processes=4
+    # - name: Report success or failure as github status
+    #   if: always()
+    #   shell: bash
+    #   run: |
+    #     status="${{ job.status }}"
+    #     lowercase_status=$(echo $status | tr '[:upper:]' '[:lower:]')
+    #     curl -sS --request POST \
+    #     --url https://api.github.com/repos/${{ github.repository }}/statuses/${{ github.sha }} \
+    #     --header 'authorization: Bearer ${{ secrets.GITHUB_TOKEN }}' \
+    #     --header 'content-type: application/json' \
+    #     --data '{
+    #         "state": "'$lowercase_status'",
+    #         "target_url": "https://github.com/${{ github.repository }}/actions/runs/${{ github.run_id }}",
+    #         "description": "'$status'",
+    #         "context": "github-actions/build"
+    #         }'