Merge pull request #491 from atomistic-machine-learning/sh/lightning_update

fix: LightningLoggerBase was removed in lightning 1.9

Author: mgastegger

requirements.txt

@@ -2,7 +2,7 @@ PyYAML
 --index-url https://pypi.python.org/simple/
 
 torch>=1.9
-pytorch_lightning>=1.8.1
+pytorch_lightning>=1.9.0
 protobuf==3.20.2
 hydra-core>=1.1.0
 hydra-colorlog>=1.1.0

setup.py

@@ -30,7 +30,7 @@ def read(fname):
         "h5py",
         "pyyaml",
         "hydra-core>=1.1.0",
-        "pytorch_lightning>=1.8.1",
+        "pytorch_lightning>=1.9.0",
         "torchmetrics",
         "hydra-colorlog>=1.1.0",
         "rich",

src/schnetpack/cli.py

@@ -10,7 +10,7 @@
 from omegaconf import DictConfig, OmegaConf
 from pytorch_lightning import LightningModule, LightningDataModule, Callback, Trainer
 from pytorch_lightning import seed_everything
-from pytorch_lightning.loggers import LightningLoggerBase
+from pytorch_lightning.loggers.logger import Logger
 
 import schnetpack as spk
 from schnetpack.utils import str2class
@@ -129,7 +129,7 @@ def train(config: DictConfig):
                 callbacks.append(hydra.utils.instantiate(cb_conf))
 
     # Init Lightning loggers
-    logger: List[LightningLoggerBase] = []
+    logger: List[Logger] = []
 
     if "logger" in config:
         for _, lg_conf in config["logger"].items():