LLMs-from-scratch/.github/workflows/basic-tests-old-pytorch.yml

name: Test PyTorch 2.2 and 2.6

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
    strategy:
      matrix:
        pytorch-version: [ 2.3.0, 2.6.0 ]

    steps:
    - uses: actions/checkout@v4

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

    - name: Install dependencies
      run: |
        curl -LsSf https://astral.sh/uv/install.sh | sh
        uv python install 3.10
        uv add . --dev
        uv pip install -r ch05/07_gpt_to_llama/tests/test-requirements-extra.txt
        uv add torch==${{ matrix.pytorch-version }}
        uv add pytest-ruff nbval

    - name: Test Selected Python Scripts
      run: |
        source .venv/bin/activate
        pytest --ruff setup/02_installing-python-libraries/tests.py
        pytest --ruff ch04/01_main-chapter-code/tests.py
        pytest --ruff ch05/01_main-chapter-code/tests.py
        pytest --ruff ch05/07_gpt_to_llama/tests/tests.py
        pytest --ruff ch06/01_main-chapter-code/tests.py

    - name: Validate Selected Jupyter Notebooks
      run: |
        source .venv/bin/activate
        pytest --ruff --nbval ch02/01_main-chapter-code/dataloader.ipynb
        pytest --ruff --nbval ch03/01_main-chapter-code/multihead-attention.ipynb
        pytest --ruff --nbval ch02/04_bonus_dataloader-intuition/dataloader-intuition.ipynb