Install jaxlib in actions

qutip · Apr 3, 2024 · 3eb253f · 3eb253f
1 parent ccb6247
commit 3eb253f
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 7 deletions.
diff --git a/.github/workflows/build_documentation.yml b/.github/workflows/build_documentation.yml
@@ -18,16 +18,17 @@ jobs:
 
       - name: Install documentation dependencies
         run: |
-          python -mpip install -r doc/requirements.txt
+          python -m pip install -r doc/requirements.txt
 
       - name: Install QuTiP from GitHub
         run: |
-          python -mpip install git+https://github.com/qutip/qutip.git@master
+          python -m pip install git+https://github.com/qutip/qutip.git@master
           python -c 'import qutip; qutip.about()'
 
       - name: Install qutip-jax from GitHub
         run: |
-          python -mpip install .[full]
+          python -m pip install jax jax[cpu]
+          python -m pip install .[full]
           python -c 'import qutip_jax; print(qutip_jax.__version__)'
 
       - name: Build documentation

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -36,15 +36,16 @@ jobs:
 
       - name: Install QuTiP from GitHub
         run: |
-          python -mpip install git+https://github.com/qutip/qutip.git@master
+          python -m pip install git+https://github.com/qutip/qutip.git@master
           python -c 'import qutip; qutip.about()'
 
       - name: Install qutip-jax and dependencies
         # Install in editable mode so Coveralls detects the tree structure
         # relative to the git repository root as well.
         run: |
-            python -mpip install -e .[full]
-            python -mpip install pytest-cov coveralls
+            python -m pip install jax jax[cpu]
+            python -m pip install -e .[full]
+            python -m pip install pytest-cov coveralls
 
       - name: Package information
         run: |

diff --git a/.github/workflows/weekly.yml b/.github/workflows/weekly.yml
@@ -51,7 +51,8 @@ jobs:
         # Install in editable mode so Coveralls detects the tree structure
         # relative to the git repository root as well.
         run: |
-            python -mpip install -e .[full]
+            python -m pip install jax jax[cpu]
+            python -m pip install -e .[full]
 
       - name: Package information
         run: |