Update CI actions

PiperOrigin-RevId: 882243703
Change-Id: Ie88ec65b49dc4bb796f927b4e9f4451a6a9ea5bc

Author: jsspencer

.github/workflows/ci-build.yaml

@@ -16,11 +16,11 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ['3.11']
+        python-version: ['3.12']
     steps:
-    - uses: actions/checkout@v4
+    - uses: actions/checkout@v6
     - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v5
+      uses: actions/setup-python@v6
       with:
         python-version: ${{ matrix.python-version }}
     - name: Install dependencies
@@ -42,13 +42,13 @@ jobs:
       matrix:
         include:
           - name-prefix: "all tests"
-            python-version: '3.11'
+            python-version: '3.12'
             os: ubuntu-latest
             package-overrides: "none"
     steps:
-    - uses: actions/checkout@v4
+    - uses: actions/checkout@v6
     - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v5
+      uses: actions/setup-python@v6
       with:
         python-version: ${{ matrix.python-version }}
     - name: Install dependencies
@@ -62,6 +62,6 @@ jobs:
       run: |
         python -m pytest
     - name: Run multi-device tests
-      if: matrix.python-version == '3.11'
+      if: matrix.python-version == '3.12'
       run: |
         FERMINET_CHEX_N_CPU_DEVICES=2 python -m pytest ferminet/tests/train_test.py