diff --git a/requirements.txt b/requirements.txt index 1ba8642f..9835e0e7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,8 +3,8 @@ 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 -omegaconf >=2.1.0 +pytorch-lightning>=1.8.0, <2.0.0 +omegaconf >=2.0.1 torchmetrics >=0.10.0, <0.12.0 tensorboard >2.2.0, !=2.5.0 protobuf >=3.20.0, <4.23.0 diff --git a/src/pytorch_tabular/tabular_model.py b/src/pytorch_tabular/tabular_model.py index 21fcdbac..7b3ec96a 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..d341ab19 100644 --- a/src/pytorch_tabular/utils.py +++ b/src/pytorch_tabular/utils.py @@ -8,7 +8,7 @@ import pandas as pd import torch import torch.nn as nn -from lightning_lite.utilities.cloud_io import get_filesystem +from pytorch_lightning.core.saving import get_filesystem from sklearn.cluster import KMeans from sklearn.preprocessing import LabelEncoder