diff --git a/requirements.txt b/requirements.txt index 1ba8642f..5c7d3fec 100644 --- a/requirements.txt +++ b/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 diff --git a/src/pytorch_tabular/tabular_model.py b/src/pytorch_tabular/tabular_model.py index 1c4a9c92..f7dcbbd7 100644 --- a/src/pytorch_tabular/tabular_model.py +++ b/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 diff --git a/src/pytorch_tabular/utils.py b/src/pytorch_tabular/utils.py index 7e0103c3..b57bf239 100644 --- a/src/pytorch_tabular/utils.py +++ b/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]