Apply isort and black reformatting

Signed-off-by: marcromeyn <[email protected]>
NVIDIA · Jul 10, 2024 · d1dad65 · d1dad65
1 parent 62b4330
commit d1dad65
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/lightning/megatron_parallel.py b/nemo/lightning/megatron_parallel.py
@@ -68,7 +68,7 @@ def default_data_step(dataloader_iter: Iterator[DataT]) -> DataT:
         )
 
     batch = next(dataloader_iter)
-    
+
     # If its wrapped in a tuple, unpack it.
     if isinstance(batch, tuple) and len(batch) == 3:
         batch = batch[0]