Python package workflow (TylerYep/torchinfo)

The Python package workflow from TylerYep/torchinfo, explained and optimized by Latchkey.

What it does

This is the Python package workflow from the TylerYep/torchinfo repository, a real project running GitHub Actions. It is shown here with attribution under its MIT license.

Below, Latchkey shows a faster, safer version produced by its optimization engine.

The workflow

name: Python package
on:
  push:
    branches: [ main ]
  pull_request:
    branches: [ main ]

permissions:
  contents: read

jobs:
  build:
    runs-on: ubuntu-latest
    strategy:
      fail-fast: false
      matrix:
        python-version: ["3.12", "3.13", "3.14"]
        pytorch-version: ["2.11"]
    steps:
    - uses: actions/checkout@v4
    - name: Install uv and set the python version
      uses: astral-sh/setup-uv@v6
      with:
        python-version: ${{ matrix.python-version }}
    - name: Install dependencies
      run: |
        uv add torch==${{ matrix.pytorch-version }}
        uv sync
    - name: mypy
      if: ${{ matrix.pytorch-version == '2.11' }}
      run: |
        uv run mypy --install-types --non-interactive .
    - name: pytest
      if: ${{ matrix.pytorch-version == '2.11' }}
      run: |
        uv run pytest --cov=torchinfo --cov-report= --durations=0
    - name: codecov
      uses: codecov/codecov-action@v5

The same workflow, on Latchkey

Removes redundant runs and caps runaway jobs. Added and changed lines are highlighted.

name: Python package
on:
  push:
    branches: [ main ]
  pull_request:
    branches: [ main ]
 
permissions:
  contents: read
 
concurrency:
  group: ${{ github.workflow }}-${{ github.ref }}
  cancel-in-progress: true
 
jobs:
  build:
    timeout-minutes: 30
    runs-on: latchkey-small
    strategy:
      fail-fast: false
      matrix:
        python-version: ["3.12", "3.13", "3.14"]
        pytorch-version: ["2.11"]
    steps:
    - uses: actions/checkout@v4
    - name: Install uv and set the python version
      uses: astral-sh/setup-uv@v6
      with:
        python-version: ${{ matrix.python-version }}
    - name: Install dependencies
      run: |
        uv add torch==${{ matrix.pytorch-version }}
        uv sync
    - name: mypy
      if: ${{ matrix.pytorch-version == '2.11' }}
      run: |
        uv run mypy --install-types --non-interactive .
    - name: pytest
      if: ${{ matrix.pytorch-version == '2.11' }}
      run: |
        uv run pytest --cov=torchinfo --cov-report= --durations=0
    - name: codecov
      uses: codecov/codecov-action@v5
 

What changed

• Run on Latchkey managed runners with one line (runs-on), which apply the fixes below automatically and self-heal transient failures. This example uses latchkey-small; pick the runner size that fits the job.
• Cancel superseded runs when a branch or PR gets a newer push.
• Add a job timeout so a hung step cannot burn hours of runner time.

2 third-party actions are referenced by a movable tag. Pin them to the commit SHA (Latchkey resolves and applies this automatically) so a repointed tag cannot change what runs.

This workflow runs 1 job (3 with the matrix expanded) per trigger. On Latchkey the same minutes cost up to 58% less than GitHub-hosted, with zero queue time.

Actions used in this workflow