PyTorch Lightning Spells

This package contains some useful plugins for PyTorch Lightning.

Disclaimer: This package is a third-party extension for PyTorch Lightning. We are not affiliated with the PyTorch Lightning project or the company behind it.

Documentation

Installation

PIP

pip install pytorch-lightning-spells

or the latest code in the main branch:

pip install https://github.com/veritable-tech/pytorch-lightning-spells/archive/main.zip

UV

uv add pytorch-lightning-spells

or the latest code in the main branch:

uv add git+https://github.com/veritable-tech/pytorch-lightning-spells.git@main

Release Notes

0.2.0 (2025-09-03)

• This release modernizes the project's tooling, dependency management, and testing infrastructure.
  • The build system has been migrated from setup.py to the standard pyproject.toml, and the CI/CD pipeline has been overhauled to use uv for significantly faster and more reliable dependency management.
  • It also introduces comprehensive tests for the callback modules, improving code reliability. It also includes several bug fixes and enhancements to existing callbacks, most notably making LookaheadModelCheckpoint more flexible and robust.
• TelegramCallback: Update the code to work with the latest version of the python-telegram-bot package.
• The GlobalMetric class has been updated to align with recent changes in torchmetrics, removing the deprecated compute_on_step parameter.