v0.1.7

What's Changed

• update code to be compatible with pytorch-lightning 2.x by @robsdavis in #9

New Contributors

• @robsdavis made their first contribution in #9

Full Changelog: v0.1.6...v0.1.7