Move from pytorch_lightning to lightning

docs/tutorials/advanced_topics/howto_pytorch_lightning.md.template

@@ -133,7 +133,7 @@ To train the model using PyTorch Lightning, we only need to extend the class wit
 
 
 ```python
-import pytorch_lightning as pl
+import lightning.pytorch as pl
 ```
 
 

docs/tutorials/advanced_topics/index.md

@@ -2,7 +2,7 @@
 
 ```{toctree}
 :maxdepth: 1
-howto_pytorch_lightning
+howto_lightning.pytorch
 hp_tuning_with_optuna
 trainer_callbacks
 ```

requirements/requirements-pytorch.txt

@@ -1,5 +1,6 @@
 torch>=1.9,<3
-pytorch-lightning>=1.5,<3
+lightning>=1.8,<2.2
+lightning.pytorch>=1.8,<2.2
 # Need to pin protobuf (for now)
 # See: https://github.com/PyTorchLightning/pytorch-lightning/issues/13159
 protobuf~=3.19.0

src/gluonts/dataset/repository/_lstnet.py

@@ -141,10 +141,7 @@ def generate_lstnet_dataset(
         pd.read_csv(ds_info.url, header=None),  # type: ignore
     )
 
-    assert df.shape == (
-        ds_info.num_time_steps,
-        ds_info.num_series,
-    ), (
+    assert df.shape == (ds_info.num_time_steps, ds_info.num_series,), (
         "expected num_time_steps/num_series"
         f" {(ds_info.num_time_steps, ds_info.num_series)} but got {df.shape}"
     )

src/gluonts/torch/model/d_linear/estimator.py

@@ -14,7 +14,7 @@
 from typing import Optional, Iterable, Dict, Any
 
 import torch
-import pytorch_lightning as pl
+import lightning.pytorch as pl
 
 from gluonts.core.component import validated
 from gluonts.dataset.common import Dataset

src/gluonts/torch/model/d_linear/lightning_module.py

@@ -11,7 +11,7 @@
 # express or implied. See the License for the specific language governing
 # permissions and limitations under the License.
 
-import pytorch_lightning as pl
+import lightning.pytorch as pl
 import torch
 
 from gluonts.core.component import validated

src/gluonts/torch/model/deepar/lightning_module.py

@@ -11,7 +11,7 @@
 # express or implied. See the License for the specific language governing
 # permissions and limitations under the License.
 
-import pytorch_lightning as pl
+import lightning.pytorch as pl
 import torch
 from torch.optim.lr_scheduler import ReduceLROnPlateau
 

src/gluonts/torch/model/estimator.py

@@ -15,7 +15,7 @@
 import logging
 
 import numpy as np
-import pytorch_lightning as pl
+import lightning.pytorch as pl
 import torch.nn as nn
 
 from gluonts.core.component import validated
@@ -217,7 +217,7 @@ def train_model(
             logger.info(
                 f"Loading best model from {checkpoint.best_model_path}"
             )
-            best_model = training_network.load_from_checkpoint(
+            best_model = training_network.__class__.load_from_checkpoint(
                 checkpoint.best_model_path
             )
         else:

src/gluonts/torch/model/lag_tst/estimator.py

@@ -14,7 +14,7 @@
 from typing import Optional, Iterable, Dict, Any, List
 
 import torch
-import pytorch_lightning as pl
+import lightning.pytorch as pl
 
 from gluonts.core.component import validated
 from gluonts.dataset.common import Dataset

src/gluonts/torch/model/lag_tst/lightning_module.py

@@ -11,7 +11,7 @@
 # express or implied. See the License for the specific language governing
 # permissions and limitations under the License.
 
-import pytorch_lightning as pl
+import lightning.pytorch as pl
 import torch
 
 from gluonts.core.component import validated

src/gluonts/torch/model/lightning_util.py

@@ -13,7 +13,7 @@
 
 from packaging import version
 
-import pytorch_lightning as pl
+import lightning.pytorch as pl
 
 
 def has_validation_loop(trainer: pl.Trainer):

src/gluonts/torch/model/mqf2/lightning_module.py

@@ -13,7 +13,7 @@
 
 from typing import Dict
 
-import pytorch_lightning as pl
+import lightning.pytorch as pl
 import torch
 from torch.optim.lr_scheduler import ReduceLROnPlateau
 

src/gluonts/torch/model/patch_tst/estimator.py

@@ -14,7 +14,7 @@
 from typing import Optional, Iterable, Dict, Any
 
 import torch
-import pytorch_lightning as pl
+import lightning.pytorch as pl
 
 from gluonts.core.component import validated
 from gluonts.dataset.common import Dataset

src/gluonts/torch/model/patch_tst/lightning_module.py

@@ -11,7 +11,7 @@
 # express or implied. See the License for the specific language governing
 # permissions and limitations under the License.
 
-import pytorch_lightning as pl
+import lightning.pytorch as pl
 import torch
 
 from gluonts.core.component import validated

src/gluonts/torch/model/simple_feedforward/estimator.py

@@ -14,7 +14,7 @@
 from typing import List, Optional, Iterable, Dict, Any
 
 import torch
-import pytorch_lightning as pl
+import lightning.pytorch as pl
 
 from gluonts.core.component import validated
 from gluonts.dataset.common import Dataset

src/gluonts/torch/model/simple_feedforward/lightning_module.py

@@ -11,7 +11,7 @@
 # express or implied. See the License for the specific language governing
 # permissions and limitations under the License.
 
-import pytorch_lightning as pl
+import lightning.pytorch as pl
 import torch
 
 from gluonts.core.component import validated

src/gluonts/torch/model/tft/lightning_module.py

@@ -11,7 +11,7 @@
 # express or implied. See the License for the specific language governing
 # permissions and limitations under the License.
 
-import pytorch_lightning as pl
+import lightning.pytorch as pl
 import torch
 from gluonts.core.component import validated
 from gluonts.itertools import select

src/gluonts/torch/model/wavenet/estimator.py

@@ -13,7 +13,7 @@
 
 from typing import Any, Dict, List, Optional, Iterable
 
-import pytorch_lightning as pl
+import lightning.pytorch as pl
 import torch
 import numpy as np
 

src/gluonts/torch/model/wavenet/lightning_module.py

@@ -11,7 +11,7 @@
 # express or implied. See the License for the specific language governing
 # permissions and limitations under the License.
 
-import pytorch_lightning as pl
+import lightning.pytorch as pl
 import torch
 
 from gluonts.core.component import validated