bump PL to <2.0

requirements.txt

@@ -3,7 +3,7 @@ category-encoders >=2.6.0, <2.7.0
 numpy >=1.17.2
 pandas >=1.1.5
 scikit-learn >=1.0.0
-pytorch-lightning >=1.8.0, <1.9.0
+pytorch-lightning>=1.8.0, <2.0.0
 omegaconf >=2.1.0
 torchmetrics >=0.10.0, <0.12.0
 tensorboard >2.2.0, !=2.5.0

src/pytorch_tabular/tabular_model.py

@@ -17,9 +17,9 @@
 import pytorch_lightning as pl
 import torch
 import torchmetrics
-from lightning_lite.utilities.seed import seed_everything
 from omegaconf import OmegaConf
 from omegaconf.dictconfig import DictConfig
+from pytorch_lightning import seed_everything
 from pytorch_lightning.callbacks import RichProgressBar
 from pytorch_lightning.callbacks.gradient_accumulation_scheduler import GradientAccumulationScheduler
 from pytorch_lightning.utilities.model_summary import summarize

src/pytorch_tabular/utils.py

@@ -8,10 +8,14 @@
 import pandas as pd
 import torch
 import torch.nn as nn
-from lightning_lite.utilities.cloud_io import get_filesystem
 from sklearn.cluster import KMeans
 from sklearn.preprocessing import LabelEncoder
 
+try:  # for 1.8
+    from pytorch_lightning.utilities.cloud_io import get_filesystem
+except ImportError:  # for 1.9
+    from pytorch_lightning.core.saving import get_filesystem
+
 import pytorch_tabular as root_module
 
 _PATH = Union[str, Path]