File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed
tests/tests_pytorch/utilities Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -42,11 +42,10 @@ def test_is_overridden():
4242 reason = "This test is ONLY relevant for the UNIFIED package" ,
4343)
4444def 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`" ):
You can’t perform that action at this time.
0 commit comments