Skip to content

Commit

Permalink
Test for PyTorch 2.6 release candidate (#500)
Browse files Browse the repository at this point in the history
* Test for PyTorch 2.6 release candidate

* update

* update

* remove extra added file
  • Loading branch information
rasbt authored Jan 23, 2025
1 parent a22d612 commit 0911e71
Show file tree
Hide file tree
Showing 2 changed files with 52 additions and 165 deletions.
52 changes: 52 additions & 0 deletions .github/workflows/basic-tests-pytorch-rc.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,52 @@
name: Test latest PyTorch release candidate
on:
push:
branches: [ main ]
paths:
- '**/*.py' # Run workflow for changes in Python files
- '**/*.ipynb'
- '**/*.yaml'
- '**/*.yml'
- '**/*.sh'
pull_request:
branches: [ main ]
paths:
- '**/*.py'
- '**/*.ipynb'
- '**/*.yaml'
- '**/*.yml'
- '**/*.sh'

jobs:
test:
runs-on: ubuntu-latest

steps:
- uses: actions/checkout@v4

- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: "3.10"

- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install pytest nbval
if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
pip install -r ch05/07_gpt_to_llama/tests/test-requirements-extra.txt
pip install torch==2.6.0 torchvision --index-url https://download.pytorch.org/whl/test/cpu
- name: Test Selected Python Scripts
run: |
pytest setup/02_installing-python-libraries/tests.py
pytest ch04/01_main-chapter-code/tests.py
pytest ch05/01_main-chapter-code/tests.py
pytest ch05/07_gpt_to_llama/tests/tests.py
pytest ch06/01_main-chapter-code/tests.py
- name: Validate Selected Jupyter Notebooks
run: |
pytest --nbval ch02/01_main-chapter-code/dataloader.ipynb
pytest --nbval ch03/01_main-chapter-code/multihead-attention.ipynb
pytest --nbval ch02/04_bonus_dataloader-intuition/dataloader-intuition.ipynb
Loading

0 comments on commit 0911e71

Please sign in to comment.