Update import path

optuna · Oct 12, 2023 · 562b3b5 · 562b3b5
1 parent 953e56d
commit 562b3b5
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/kubernetes/mlflow/pytorch_lightning_distributed.py b/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

diff --git a/pytorch/pytorch_lightning_ddp.py b/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

diff --git a/pytorch/pytorch_lightning_simple.py b/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