tests

anas-rz · Feb 28, 2024 · 54ec5bb · 54ec5bb
1 parent fc4084c
commit 54ec5bb
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 2 deletions.
diff --git a/.github/workflows/test_jax.yml b/.github/workflows/test_jax.yml
@@ -27,6 +27,7 @@ jobs:
     - name: Install dependencies
       run: |
         pip3 install -r requirements/requirements_test.txt
+        pip install keras --upgrade
 
     - name: Test with pytest and coverage
       run: |

diff --git a/.github/workflows/test_tensorflow.yml b/.github/workflows/test_tensorflow.yml
@@ -27,6 +27,7 @@ jobs:
     - name: Install dependencies
       run: |
         pip install -r requirements/requirements_test.txt
+        pip install keras --upgrade
 
     - name: Test with pytest and coverage
       run: |

diff --git a/.github/workflows/test_torch.yml b/.github/workflows/test_torch.yml
@@ -26,8 +26,8 @@ jobs:
         python -m pip install pytest coverage codecov
     - name: Install dependencies
       run: |
-        pip3 install torch --index-url https://download.pytorch.org/whl/cpu
         pip3 install -r requirements/requirements_test.txt
+        pip install keras --upgrade
 
     - name: Test with pytest and coverage
       run: |

diff --git a/requirements/requirements_test.txt b/requirements/requirements_test.txt
@@ -1,3 +1,3 @@
-torchmetrics
+torch --index-url https://download.pytorch.org/whl/cpu
 tensorflow
 jax[cpu]