[CI] Use Python 3.12 as default for tests (#632)

* [CI] Use Python 3.12 as default for tests

* Add Python 3.12 to the Linux test matrix.
* Use Python 3.12 as the default Python for all other CI tests.

* [CI] Remove jax upper bounds for testing

* reset caps on jax

* Update requirements_all.txt

* Update setup.py

* Update setup.py

* Update requirements_all.txt

* Update requirements_all.txt

* Update requirements_all.txt

---------

Co-authored-by: Cédric Vincent-Cuaz <cedvincentcuaz@gmail.com>

Author: matthewfeickert

.github/workflows/build_tests.yml

@@ -4,10 +4,10 @@ on:
   workflow_dispatch:
   pull_request:
     branches:
-      - 'master'    
+      - 'master'
   push:
     branches:
-      - 'master'  
+      - 'master'
   create:
     branches:
       - 'master'
@@ -22,7 +22,7 @@ jobs:
     strategy:
       max-parallel: 4
       matrix:
-        python-version: ["3.8", "3.9", "3.10", "3.11"]
+        python-version: ["3.8", "3.9", "3.10", "3.11", "3.12"]
 
     steps:
     - uses: actions/checkout@v4
@@ -49,14 +49,14 @@ jobs:
     if: "!contains(github.event.head_commit.message, 'no pep8')"
     steps:
     - uses: actions/checkout@v4
-    - name: Set up Python 
+    - name: Set up Python
       uses: actions/setup-python@v5
       with:
-        python-version: "3.10"
+        python-version: "3.x"
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip setuptools
-        pip install flake8 
+        pip install flake8
     - name: Lint with flake8
       run: |
         # stop the build if there are Python syntax errors or undefined names
@@ -70,10 +70,10 @@ jobs:
     if: "!contains(github.event.head_commit.message, 'no ci')"
     steps:
     - uses: actions/checkout@v4
-    - name: Set up Python 
+    - name: Set up Python
       uses: actions/setup-python@v5
       with:
-        python-version: "3.10"
+        python-version: "3.12"
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip setuptools
@@ -94,7 +94,7 @@ jobs:
        max-parallel: 4
        matrix:
          os: [macos-latest, macos-13]
-         python-version: ["3.11"]
+         python-version: ["3.12"]
 
      steps:
      - uses: actions/checkout@v4
@@ -109,7 +109,7 @@ jobs:
        run: |
          python -m pip install --upgrade pip setuptools
          pip install -r requirements_all.txt
-         pip install pytest 
+         pip install pytest
      - name: Run tests
        run: |
          python -m pytest --durations=20  -v test/ ot/ --color=yes
@@ -121,7 +121,7 @@ jobs:
     strategy:
       max-parallel: 4
       matrix:
-        python-version: ["3.11"]
+        python-version: ["3.12"]
 
     steps:
     - uses: actions/checkout@v4

requirements_all.txt

@@ -2,7 +2,7 @@ numpy>=1.20
 scipy>=1.6
 matplotlib
 autograd
-pymanopt
+pymanopt @ git+https://github.com/pymanopt/pymanopt.git@master
 cvxopt
 scikit-learn
 torch