Make GraphNeT compatible with pytorch-lightning >= 2.0

Created by: asogaard

Currently there are some breaking changes in pytorch-lightning 2.0 which mean that GraphNeT doesn't work with it (see #470). It would be nice to be compatible with this newest version.