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