add checker

kornia · Jan 26, 2024 · 6141057 · 6141057
1 parent c964b1a
commit 6141057
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/.github/actions/env/action.yml b/.github/actions/env/action.yml
@@ -43,7 +43,11 @@ runs:
       shell: bash
       run: pip install .[dev,x]
 
-    - name: Check dependencies and kornia version
+    - name: Check torch version
+      shell: bash
+      run: pip show torch | grep ${{ inputs.pytorch-version }}  || false
+
+    - name: Print dependencies and kornia version
       shell: bash
       run: |
         python -c "import torch;print('Pytorch version: ', torch.__version__)"

diff --git a/.github/workflows/self-tests.yml b/.github/workflows/self-tests.yml
@@ -23,6 +23,7 @@ jobs:
         path: subdir
     - uses: ./subdir/.github/actions/env/
       with:
+        pytorch-version: '1.9.1'
         ref: main
 
   coverage: