update

Author: Borda

src/lightning/pytorch/plugins/io/async_plugin.py

@@ -13,15 +13,15 @@
 # limitations under the License.
 
 from concurrent.futures import ThreadPoolExecutor
-from typing import Any, Optional
+from typing import Any, Optional, TYPE_CHECKING
 
 import torch
 from lightning_utilities.core.apply_func import apply_to_collection
 from typing_extensions import override
 
 from lightning.pytorch.plugins.io.wrapper import _WrappingCheckpointIO
 
-if TYPING_CHE:
+if TYPE_CHECKING:
     from lightning.fabric.plugins import CheckpointIO
 
 
@@ -86,6 +86,7 @@ def teardown(self) -> None:
         if self._error:
             raise self._error
 
+
 # snapshot the checkpoint payload on the caller thread to avoid races with parameter mutation
 def _clone_tensor(t: torch.Tensor) -> torch.Tensor:
     # detach to avoid autograd history and clone to take a point-in-time copy