Merge pull request #213 from Alnusjaponica/fix-lightning-import-path

Update import path

kubernetes/mlflow/pytorch_lightning_distributed.py

@@ -1,9 +1,9 @@
 import os
 
+import lightning.pytorch as pl
+from lightning.pytorch import Callback
 import optuna
 from optuna.integration.mlflow import MLflowCallback
-import pytorch_lightning as pl
-from pytorch_lightning import Callback
 import torch
 import torch.nn as nn
 import torch.nn.functional as F

pytorch/pytorch_lightning_ddp.py

@@ -16,9 +16,9 @@
 from typing import List
 from typing import Optional
 
+import lightning.pytorch as pl
 import optuna
 from optuna.integration import PyTorchLightningPruningCallback
-import pytorch_lightning as pl
 import torch
 from torch import nn
 from torch import optim

pytorch/pytorch_lightning_simple.py

@@ -15,10 +15,10 @@
 from typing import List
 from typing import Optional
 
+import lightning.pytorch as pl
 import optuna
 from optuna.integration import PyTorchLightningPruningCallback
 from packaging import version
-import pytorch_lightning as pl
 import torch
 from torch import nn
 from torch import optim

pytorch/requirements.txt

@@ -1,7 +1,7 @@
 mpi4py
 plotly
 pytorch-ignite
-pytorch-lightning
+lightning
 skorch
 torch
 torchvision