diff --git a/.github/actions/env/action.yml b/.github/actions/env/action.yml index f9ea68a..a9485e6 100644 --- 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 index 90bf884..e4a0620 100644 --- 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: