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 898195c + 13383e0 commit f4fab99Copy full SHA for f4fab99
tests/tests_pytorch/utilities/test_compile.py
@@ -13,13 +13,12 @@
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
22
-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
23
from lightning.pytorch import LightningModule, Trainer
24
from lightning.pytorch.demos.boring_classes import BoringModel
25
from lightning.pytorch.utilities.compile import from_compiled, to_uncompiled
0 commit comments