We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 1837dbd + 123a122 commit dd19f16Copy full SHA for dd19f16
tests/tests_pytorch/utilities/test_compile.py
@@ -13,14 +13,13 @@
13
# limitations under the License.
14
import os
15
import sys
16
-from contextlib import nullcontext
17
from unittest import mock
18
19
import pytest
20
import torch
21
from lightning_utilities.core.imports import RequirementCache
22
23
-from lightning.fabric.utilities.imports import _TORCH_GREATER_EQUAL_2_2, _TORCH_GREATER_EQUAL_2_4
+from lightning.fabric.utilities.imports import _TORCH_GREATER_EQUAL_2_2
24
from lightning.pytorch import LightningModule, Trainer
25
from lightning.pytorch.demos.boring_classes import BoringModel
26
from lightning.pytorch.utilities.compile import from_compiled, to_uncompiled
0 commit comments