Merge branch 'gcassella/update-kfac-api' of https://github.com/gcasse…

…lla/ferminet-1 into gcassella/update-kfac-api
google-deepmind · Nov 6, 2023 · 548d93d · 548d93d
2 parents d5c8966 + 313d0ee
commit 548d93d
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/.github/workflows/ci-build.yaml b/.github/workflows/ci-build.yaml
@@ -16,7 +16,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: [3.8]
+        python-version: ['3.10']
 
     steps:
     - uses: actions/checkout@v2
@@ -55,15 +55,11 @@ jobs:
       matrix:
         include:
           - name-prefix: "all tests"
-            python-version: 3.8
-            os: ubuntu-latest
-            package-overrides: "none"
-          - name-prefix: "all tests"
-            python-version: 3.9
+            python-version: '3.10'
             os: ubuntu-latest
             package-overrides: "none"
           - name-prefix: "with internal numpy"
-            python-version: 3.9
+            python-version: '3.10'
             os: ubuntu-latest
             # Test with numpy version that matches Google-internal version
             package-overrides: "numpy==1.21.5"
@@ -95,6 +91,6 @@ jobs:
       run: |
         python -m pytest
     - name: Run multi-device tests
-      if: matrix.python-version == 3.8
+      if: matrix.python-version == '3.10'
       run: |
         FERMINET_CHEX_N_CPU_DEVICES=2 python -m pytest ferminet/tests/train_test.py