Skip to content

Commit 3e7ec3a

Browse files
committed
run pre-commit
1 parent 156452b commit 3e7ec3a

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/tests_pytorch/utilities/test_model_helpers.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -42,11 +42,10 @@ def test_is_overridden():
4242
reason="This test is ONLY relevant for the UNIFIED package",
4343
)
4444
def test_mixed_imports_unified():
45-
from pytorch_lightning.callbacks import EarlyStopping as OldEarlyStopping
46-
from pytorch_lightning.demos.boring_classes import BoringModel as OldBoringModel
47-
4845
from lightning.pytorch.utilities.compile import _maybe_unwrap_optimized as new_unwrap
4946
from lightning.pytorch.utilities.model_helpers import is_overridden as new_is_overridden
47+
from pytorch_lightning.callbacks import EarlyStopping as OldEarlyStopping
48+
from pytorch_lightning.demos.boring_classes import BoringModel as OldBoringModel
5049

5150
model = OldBoringModel()
5251
with pytest.raises(TypeError, match=r"`pytorch_lightning` object \(BoringModel\) to a `lightning.pytorch`"):

0 commit comments

Comments
 (0)