Skip to content

Notebook for Learning Rate Scheduler #710

Notebook for Learning Rate Scheduler

Notebook for Learning Rate Scheduler #710

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.9', '3.10', '3.11', '3.12']
torch_version: ['2.2.2+cpu', '2.3.1+cpu', '2.4.1+cpu', '2.5.1+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