looks like an unrequired import got added to native_amp sometime in t…

…he last few hours
Lightning-AI · Oct 5, 2022 · bba3e82 · bba3e82
1 parent 635d9bd
commit bba3e82
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pytorch_lightning/plugins/precision/native_amp.py b/src/pytorch_lightning/plugins/precision/native_amp.py
@@ -20,7 +20,7 @@
 
 import pytorch_lightning as pl
 from lightning_lite.accelerators.cuda import _patch_cuda_is_available
-from lightning_lite.utilities.types import Optimizable, Steppable
+from lightning_lite.utilities.types import Optimizable
 from pytorch_lightning.plugins.precision.precision_plugin import PrecisionPlugin
 from pytorch_lightning.utilities import _TORCH_GREATER_EQUAL_1_10, AMPType
 from pytorch_lightning.utilities.exceptions import MisconfigurationException