Apply isort and black reformatting

Signed-off-by: suiyoubi <suiyoubi@users.noreply.github.com>
NVIDIA · Nov 7, 2024 · 819899d · 819899d
1 parent 1768003
commit 819899d
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/nemo/lightning/pytorch/strategies/megatron_strategy.py b/nemo/lightning/pytorch/strategies/megatron_strategy.py
@@ -871,8 +871,7 @@ def current_epoch_step(self) -> int:
 
     @property
     def distributed_sampler_kwargs(self) -> Dict[str, Any]:
-        """ Get distributed sampler kwargs.
-        """
+        """Get distributed sampler kwargs."""
         from nemo.utils import AppState
 
         app_state = AppState()
@@ -898,8 +897,7 @@ def restore_checkpoint_after_setup(self) -> bool:
 
     @property
     def parallelism(self) -> ParallelismConfig:
-        """ Get ParallelismConfig.
-        """
+        """Get ParallelismConfig."""
         return ParallelismConfig(
             tensor_model_parallel_size=self.tensor_model_parallel_size,
             pipeline_model_parallel_size=self.pipeline_model_parallel_size,