cache testing resources on version and dep changes

Signed-off-by: Jeffrey Martin <[email protected]>

Author: jmartin-tech

.github/workflows/test_linux.yml

@@ -35,11 +35,38 @@ jobs:
         uses: actions/setup-python@v4
         with:
           python-version: ${{ matrix.python-version }}
+
+      - name: Restore test cache artifacts
+        id: cache-artifacts-restore
+        uses: actions/cache/restore@v4
+        with:
+          path: |
+            ~/.cache/garak/data
+            ~/.cache/huggingface
+          key: ${{ runner.os }}-${{ hashFiles('**/pyproject.toml') }}
+
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
           pip install -r requirements.txt
           python -m pip cache purge
+
       - name: Test with pytest
         run: |
           python -m pytest tests/
+
+      - name: Prepare resources for cache
+        run: |
+          rm -rf ~/.cache/huggingface/hub/*facebook*
+          rm -rf ~/.cache/huggingface/hub/*Helsinki*
+
+      - name: Save test cache
+        if: steps.cache-artifacts-restore.outputs.cache-hit != 'true'
+        uses: actions/cache@v4
+        with:
+          path: |
+            ~/.cache/garak/data
+            ~/.cache/huggingface
+            !~/.cache/huggingface/hub/*facebook*
+            !~/.cache/huggingface/hub/*Helsinki*
+          key: ${{ runner.os }}-${{ hashFiles('**/pyproject.toml') }}

.github/workflows/test_macos.yml

@@ -39,6 +39,15 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
 
+      - name: Restore test cache artifacts
+        id: cache-artifacts-restore
+        uses: actions/cache/restore@v4
+        with:
+          path: |
+            ~/.cache/garak/data
+            ~/.cache/huggingface
+          key: ${{ runner.os }}-${{ hashFiles('**/pyproject.toml') }}
+
       - name: Install dependencies
         run: |
           brew install libmagic
@@ -51,3 +60,17 @@ jobs:
         run: |
           cd garak
           python -m pytest tests/
+
+      - name: Prepare resources for cache
+        run: |
+          rm -rf ~/.cache/huggingface/hub/*facebook*
+          rm -rf ~/.cache/huggingface/hub/*Helsinki*
+
+      - name: Save test cache
+        if: steps.cache-artifacts-restore.outputs.cache-hit != 'true'
+        uses: actions/cache@v4
+        with:
+          path: |
+            ~/.cache/garak/data
+            ~/.cache/huggingface
+          key: ${{ runner.os }}-${{ hashFiles('**/pyproject.toml') }}

.github/workflows/test_windows.yml

@@ -39,6 +39,15 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
 
+      - name: Restore test cache artifacts
+        id: cache-artifacts-restore
+        uses: actions/cache/restore@v4
+        with:
+          path: |
+            ~/.cache/garak/data
+            ~/.cache/huggingface
+          key: ${{ runner.os }}-${{ hashFiles('**/pyproject.toml') }}
+
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
@@ -50,3 +59,19 @@ jobs:
         run: |
           cd garak
           python -m pytest tests/
+
+      - name: Prepare resources for cache
+        run: |
+          rm -rf ~/.cache/huggingface/hub/*facebook*
+          rm -rf ~/.cache/huggingface/hub/*Helsinki*
+
+      - name: Save test cache
+        if: steps.cache-artifacts-restore.outputs.cache-hit != 'true'
+        uses: actions/cache@v4
+        with:
+          path: |
+            ~/.cache/garak/data
+            ~/.cache/huggingface
+            !~/.cache/huggingface/hub/*facebook*
+            !~/.cache/huggingface/hub/*Helsinki*
+          key: ${{ runner.os }}-${{ hashFiles('**/pyproject.toml') }}