Remove unused tensorstore import

NVIDIA · Jun 19, 2024 · 2681ab3 · 2681ab3
1 parent cb27c06
commit 2681ab3
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/nemo/lightning/io/pl.py b/nemo/lightning/io/pl.py
@@ -8,7 +8,6 @@
 from lightning_fabric.plugins.io.checkpoint_io import CheckpointIO
 from lightning_fabric.utilities.cloud_io import get_filesystem
 from lightning_fabric.utilities.types import _PATH
-from megatron.core.dist_checkpointing.strategies import tensorstore
 from torch import nn
 from typing_extensions import Self, override