Skip to content

Commit deb7378

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

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/tests_pytorch/utilities/test_model_helpers.py

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

48+
from lightning.pytorch.utilities.compile import _maybe_unwrap_optimized as new_unwrap
49+
from lightning.pytorch.utilities.model_helpers import is_overridden as new_is_overridden
50+
5051
model = OldBoringModel()
5152
with pytest.raises(TypeError, match=r"`pytorch_lightning` object \(BoringModel\) to a `lightning.pytorch`"):
5253
new_unwrap(model)

0 commit comments

Comments
 (0)