Skip to content

Add PyTorch 2.4.0 to CI (#1063) #690

Add PyTorch 2.4.0 to CI (#1063)

Add PyTorch 2.4.0 to CI (#1063) #690

Workflow file for this run

# This workflow will install Python dependencies, run tests and lint with a single version of Python
# For more information see: https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions
name: tests
on:
push:
branches: [ master ]
pull_request:
branches: [ master ]
# print more columns
env:
COLUMNS: 120
jobs:
test:
runs-on: ${{ matrix.os }}
strategy:
fail-fast: false # don't cancel all jobs when one fails
matrix:
python_version: ['3.8', '3.9', '3.10', '3.11']
torch_version: ['2.1.2+cpu', '2.2.2+cpu', '2.3.1+cpu', '2.4.0+cpu']
os: [ubuntu-latest]
steps:
- uses: actions/checkout@v3
- name: Set up Python
uses: actions/setup-python@v4
with:
python-version: ${{ matrix.python_version }}
- name: Install dependencies
# TODO remove numpy version constraint once we no longer support PyTorch < 2.3
run: |
python -m pip install --upgrade pip
python -m pip install -r requirements-dev.txt
python -m pip install -r requirements.txt
python -m pip install --force-reinstall -U "numpy<2.0.0"
python -m pip install pytest-pretty
python -m pip install torch==${{ matrix.torch_version }} -f https://download.pytorch.org/whl/torch
python -m pip list
- name: Install skorch
run: |
python -m pip install .
- name: Test with pytest
run: |
pytest