Apply isort and black reformatting

Signed-off-by: ashors1 <[email protected]>
NVIDIA · Aug 2, 2024 · fbd55b9 · fbd55b9
1 parent b9306e8
commit fbd55b9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/lightning/test_dist_ckpt.py b/tests/lightning/test_dist_ckpt.py
@@ -6,12 +6,13 @@
 import pytest
 import pytorch_lightning as pl
 import torch
+from megatron.core.num_microbatches_calculator import reconfigure_num_microbatches_calculator
 
 import nemo.lightning as nl
 from nemo.collections import llm
 from nemo.lightning.io.pl import MegatronCheckpointIO
 from nemo.utils.callbacks.dist_ckpt_io import AsyncFinalizableCheckpointIO, AsyncFinalizerCallback
-from megatron.core.num_microbatches_calculator import reconfigure_num_microbatches_calculator
+
 
 def _get_strategy():
     strategy = nl.MegatronStrategy(