pytorch-lightning

a25515e9 - CI: fix doctest failure from PyTorch LeafSpec FutureWarning (#21502)