Skip to content

Commit 19fe9bb

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

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

tests/tests_pytorch/graveyard/test_precision.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,11 @@
33

44
def test_precision_plugin_renamed_imports():
55
# base class
6+
from lightning.pytorch.plugins.precision.precision_plugin import PrecisionPlugin as PrecisionPlugin0
7+
68
from lightning.pytorch.plugins import PrecisionPlugin as PrecisionPlugin2
79
from lightning.pytorch.plugins.precision import PrecisionPlugin as PrecisionPlugin1
810
from lightning.pytorch.plugins.precision.precision import Precision
9-
from lightning.pytorch.plugins.precision.precision_plugin import PrecisionPlugin as PrecisionPlugin0
1011

1112
assert issubclass(PrecisionPlugin0, Precision)
1213
assert issubclass(PrecisionPlugin1, Precision)

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)