isort fix

Signed-off-by: Zeeshan Patel <[email protected]>
NVIDIA · Nov 9, 2024 · 654b7b6 · 654b7b6
1 parent f477c30
commit 654b7b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/diffusion/train.py b/nemo/collections/diffusion/train.py
@@ -40,9 +40,9 @@
 )
 from nemo.collections.multimodal.data.energon.base import SimpleMultiModalDataModule
 from nemo.lightning.pytorch.callbacks import ModelCheckpoint, PreemptionCallback
-from nemo.lightning.pytorch.callbacks.nsys import NsysCallback
 from nemo.lightning.pytorch.callbacks.megatron_comm_overlap import MegatronCommOverlapCallback
 from nemo.lightning.pytorch.callbacks.model_transform import ModelTransform
+from nemo.lightning.pytorch.callbacks.nsys import NsysCallback
 from nemo.lightning.pytorch.strategies.utils import RestoreConfig
 from nemo.utils.exp_manager import TimingCallback