Skip to content

Commit dadb6c9

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

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

src/lightning/pytorch/callbacks/memory_format.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,8 @@
1818
changes the model memory format
1919
"""
2020

21-
from typing import Any, MutableSequence, Optional
21+
from collections.abc import MutableSequence
22+
from typing import Any, Optional
2223

2324
import torch
2425

tests/tests_pytorch/callbacks/test_memory_format.py

+1
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
from unittest.mock import MagicMock
1616

1717
import torch
18+
1819
from lightning.pytorch import LightningModule, Trainer
1920
from lightning.pytorch.callbacks import MemoryFormat
2021

0 commit comments

Comments
 (0)