Merge branch 'master' into fix-mathjax-docs

Author: mhucka

.github/workflows/ci-build-checks.yaml

@@ -24,38 +24,52 @@ on:
     types:
       - checks_requested
 
+concurrency:
+  cancel-in-progress: true
+  group: ${{github.workflow}}-${{github.event.pull_request.number||github.ref}}
+
 permissions: read-all
 
 jobs:
   wheel-build:
     name: Wheel test
     runs-on: ubuntu-22.04
+    timeout-minutes: 60
 
     steps:
-      - uses: actions/checkout@544eadc6bf3d226fd7a7a9f0dc5b5bf7ca0675b9 # v1
-      - uses: actions/setup-python@0f07f7f756721ebd886c2462646a35f78a8bc4de # v1
+      - uses: actions/checkout@1af3b93b6815bc44a9784bd300feb67ff0d1eeb3 # v6.0.0
+      - uses: actions/setup-python@e797f83bcb11b83ae66e0230d6156d7c80228e7c # v5
         with:
           python-version: '3.10'
           architecture: 'x64'
+          cache: pip
       - name: Install Bazel on CI
         run: ./scripts/ci_install.sh
-      - name: Configure CI TF
-        run: echo "Y\n" | ./configure.sh
       - name: Build Wheel Test
         run: ./scripts/build_pip_package_test.sh
       - name: Test Wheel
         run: ./scripts/run_example.sh
 
   bazel-tests:
     name: Library tests
-    runs-on: ubuntu-22.04
+    runs-on: linux-x86-n2-32
+    container:
+      image: us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build:latest
+    timeout-minutes: 60
 
     steps:
-      - uses: actions/checkout@544eadc6bf3d226fd7a7a9f0dc5b5bf7ca0675b9 # v1
-      - uses: actions/setup-python@0f07f7f756721ebd886c2462646a35f78a8bc4de # v1
+      - name: Adjust the ml-build runner to our needs
+        run: |
+          # lsb-release (needed by setup-python) is missing in ml-build runners
+          sudo apt-get update && sudo apt-get install -y lsb-release
+          # This file, though it's empty, hangs the Bazel install step.
+          rm -f /etc/bazel.bazelrc
+      - uses: actions/checkout@1af3b93b6815bc44a9784bd300feb67ff0d1eeb3 # v6.0.0
+      - uses: actions/setup-python@e797f83bcb11b83ae66e0230d6156d7c80228e7c # v5
         with:
           python-version: '3.10'
           architecture: 'x64'
+          cache: pip
       - name: Install Bazel on CI
         run: ./scripts/ci_install.sh
       - name: Configure CI TF
@@ -87,21 +101,28 @@ jobs:
 
   tutorials-test:
     name: Tutorial tests
-    runs-on: ubuntu-22.04
-    needs: wheel-build
+    runs-on: linux-x86-n2-32
+    container:
+      image: us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build:latest
+    timeout-minutes: 60
 
     steps:
-      - uses: actions/checkout@544eadc6bf3d226fd7a7a9f0dc5b5bf7ca0675b9 # v1
-      - uses: actions/setup-python@0f07f7f756721ebd886c2462646a35f78a8bc4de # v1
+      - name: Adjust the ml-build runner to our needs
+        run: |
+          # lsb-release (needed by setup-python) is missing in ml-build runners
+          sudo apt-get update && sudo apt-get install -y lsb-release
+          # This file, though it's empty, hangs the Bazel install step.
+          rm -f /etc/bazel.bazelrc
+      - uses: actions/checkout@1af3b93b6815bc44a9784bd300feb67ff0d1eeb3 # v6.0.0
+      - uses: actions/setup-python@e797f83bcb11b83ae66e0230d6156d7c80228e7c # v5
         with:
           python-version: '3.10'
           architecture: 'x64'
+          cache: pip
       - name: Install notebook dependencies
         run: pip install --upgrade pip seaborn==0.10.0
       - name: Install Bazel on CI
         run: ./scripts/ci_install.sh
-      - name: Configure CI TF
-        run: echo "Y\n" | ./configure.sh
       - name: Build Wheel
         run: ./scripts/build_pip_package_test.sh
       - name: Test Notebooks

tensorflow_quantum/python/optimizers/BUILD

@@ -39,6 +39,7 @@ py_test(
 py_test(
     name = "spsa_minimizer_test",
     srcs = ["spsa_minimizer_test.py"],
+    flaky = True,
     python_version = "PY3",
     deps = [
         ":spsa_minimizer",