Merge pull request #26 from kornia/feat/cached-weights

feat: cache models weights tests
kornia · Aug 14, 2024 · 9a83cbf · 9a83cbf
2 parents 30598a5 + 8060b2c
commit 9a83cbf
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -33,6 +33,15 @@ on:
         required: false
         type: string
         default: ${{ github.sha }}
+      cache-path:
+        required: false
+        type: string
+      cache-key:
+        required: false
+        type: string
+      cache-restore-keys:
+        required: false
+        type: string
 
 jobs:
   tests:
@@ -77,6 +86,14 @@ jobs:
             python-version: '3.12'
 
     steps:
+    - name: Restore cache
+      if: github.event.inputs.cache-path != ''
+      uses: actions/cache/restore@v4
+      with:
+        path: ${{ inputs.cache-path }}
+        key: ${{ inputs.cache-key }}
+        restore-keys: ${{ inputs.cache-restore-keys }}
+
     - uses: kornia/workflows/.github/actions/[email protected]
       with:
         os: ${{ inputs.os }}