Update import from lightning.pytorch (#409)

* update import * Remove unnecessary import return type --------- Co-authored-by: Filippo Olivo <filippo@filippoolivo.com>
2025-01-22 10:48:42 +01:00
parent d51de028bd
commit 4bec5bfc9a
4 changed files with 6 additions and 7 deletions
--- a/pina/callbacks/adaptive_refinment_callbacks.py
+++ b/pina/callbacks/adaptive_refinment_callbacks.py
@@ -1,7 +1,7 @@
 """PINA Callbacks Implementations"""

 import torch
-from pytorch_lightning.callbacks import Callback
+from lightning.pytorch.callbacks import Callback
 from ..label_tensor import LabelTensor
 from ..utils import check_consistency