Skip to content

Commit a214154

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 74cf6a7 commit a214154

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

src/lightning/pytorch/callbacks/model_checkpoint.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,9 +30,12 @@
3030
from typing import Any, Literal, Optional, Union, cast
3131
from weakref import proxy
3232

33-
import pytorch_lightning as pl
3433
import torch
3534
import yaml
35+
from torch import Tensor
36+
from typing_extensions import override
37+
38+
import pytorch_lightning as pl
3639
from lightning_fabric.utilities.cloud_io import (
3740
_is_dir,
3841
_is_local_file_protocol,
@@ -47,8 +50,6 @@
4750
rank_zero_warn,
4851
)
4952
from pytorch_lightning.utilities.types import STEP_OUTPUT
50-
from torch import Tensor
51-
from typing_extensions import override
5253

5354
log = logging.getLogger(__name__)
5455
warning_cache = WarningCache()

tests/tests_pytorch/checkpointing/test_model_checkpoint.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,6 @@
3232
from torch import optim
3333
from torch.utils.data.dataloader import DataLoader
3434

35-
import lightning.pytorch as pl
3635
from lightning.fabric.utilities.cloud_io import _load as pl_load
3736
from lightning.pytorch import Trainer, seed_everything
3837
from lightning.pytorch.callbacks import ModelCheckpoint

0 commit comments

Comments
 (0)