Fix CI

james77777778 · Jul 25, 2024 · 4122d9c · 4122d9c
1 parent 999e8fd
commit 4122d9c
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 7 deletions.
diff --git a/.github/workflows/actions.yml b/.github/workflows/actions.yml
@@ -32,10 +32,10 @@ jobs:
         uses: actions/cache@v4
         with:
           path: ${{ steps.pip-cache.outputs.dir }}
-          key: ${{ runner.os }}-pip-${{ hashFiles('pyproject.toml') }}-${{ hashFiles('requirements.txt') }}
+          key: ${{ runner.os }}-pip-${{ hashFiles('pyproject.toml') }}-${{ hashFiles('requirements_ci.txt') }}
       - name: Install dependencies
         run: |
-          pip install -r requirements.txt --progress-bar off --upgrade
+          pip install -r requirements_ci.txt --progress-bar off --upgrade
           pip install -e ".[tests]" --progress-bar off --upgrade
       - name: Check for API changes
         run: |
@@ -71,10 +71,10 @@ jobs:
         uses: actions/cache@v4
         with:
           path: ${{ steps.pip-cache.outputs.dir }}
-          key: ${{ runner.os }}-pip-${{ hashFiles('pyproject.toml') }}-${{ hashFiles('requirements.txt') }}
+          key: ${{ runner.os }}-pip-${{ hashFiles('pyproject.toml') }}-${{ hashFiles('requirements_ci.txt') }}
       - name: Install dependencies
         run: |
-          pip install -r requirements.txt --progress-bar off --upgrade
+          pip install -r requirements_ci.txt --progress-bar off --upgrade
           pip install -e ".[tests]" --progress-bar off --upgrade
       - name: Test with pytest
         run: |

diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -25,10 +25,10 @@ jobs:
         uses: actions/cache@v4
         with:
           path: ${{ steps.pip-cache.outputs.dir }}
-          key: ${{ runner.os }}-pip-${{ hashFiles('pyproject.toml') }}-${{ hashFiles('requirements.txt') }}
+          key: ${{ runner.os }}-pip-${{ hashFiles('pyproject.toml') }}-${{ hashFiles('requirements_ci.txt') }}
       - name: Install dependencies
         run: |
-          pip install -r requirements.txt --progress-bar off --upgrade
+          pip install -r requirements_ci.txt --progress-bar off --upgrade
           pip install -e ".[tests]" --progress-bar off --upgrade
       - name: Check for API changes
         run: |

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
-# THIS IS FOR GRADIO APP ONLY
+# For Gradio app
 --extra-index-url https://download.pytorch.org/whl/cpu
 torch
 torchvision

diff --git a/requirements_ci.txt b/requirements_ci.txt
@@ -0,0 +1,10 @@
+# For CI
+tensorflow-cpu
+
+--extra-index-url https://download.pytorch.org/whl/cpu
+torch
+torchvision
+
+jax[cpu]
+
+keras