From 715a3db8ddd1b61e075eed86932a8042d03e5cdc Mon Sep 17 00:00:00 2001
From: akoumpa <akoumpa@users.noreply.github.com>
Date: Tue, 13 Aug 2024 21:37:37 +0000
Subject: [PATCH] Apply isort and black reformatting

Signed-off-by: akoumpa <akoumpa@users.noreply.github.com>
---
 nemo/lightning/__init__.py                  | 2 +-
 nemo/lightning/pytorch/plugins/__init__.py  | 2 +-
 nemo/lightning/pytorch/plugins/precision.py | 3 +++
 3 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/nemo/lightning/__init__.py b/nemo/lightning/__init__.py
index 2f8b2707836a..cc2f624a9ba3 100644
--- a/nemo/lightning/__init__.py
+++ b/nemo/lightning/__init__.py
@@ -16,7 +16,7 @@
 from nemo.lightning.nemo_logger import NeMoLogger
 from nemo.lightning.pytorch.callbacks.model_checkpoint import ModelCheckpoint
 from nemo.lightning.pytorch.optim import LRSchedulerModule, MegatronOptimizerModule, OptimizerModule, lr_scheduler
-from nemo.lightning.pytorch.plugins import MegatronDataSampler, MegatronMixedPrecision, MegatronCustomPrecision
+from nemo.lightning.pytorch.plugins import MegatronCustomPrecision, MegatronDataSampler, MegatronMixedPrecision
 from nemo.lightning.pytorch.plugins import data_sampler as _data_sampler
 from nemo.lightning.pytorch.strategies import MegatronStrategy
 from nemo.lightning.pytorch.trainer import Trainer
diff --git a/nemo/lightning/pytorch/plugins/__init__.py b/nemo/lightning/pytorch/plugins/__init__.py
index e886f21acaf5..d6cdf573c782 100644
--- a/nemo/lightning/pytorch/plugins/__init__.py
+++ b/nemo/lightning/pytorch/plugins/__init__.py
@@ -1,5 +1,5 @@
 from nemo.lightning.pytorch.plugins.data_sampler import MegatronDataSampler
-from nemo.lightning.pytorch.plugins.precision import MegatronMixedPrecision, MegatronCustomPrecision
+from nemo.lightning.pytorch.plugins.precision import MegatronCustomPrecision, MegatronMixedPrecision
 
 __all__ = [
     "MegatronDataSampler",
diff --git a/nemo/lightning/pytorch/plugins/precision.py b/nemo/lightning/pytorch/plugins/precision.py
index c0f21c92bd81..bf148235ae10 100644
--- a/nemo/lightning/pytorch/plugins/precision.py
+++ b/nemo/lightning/pytorch/plugins/precision.py
@@ -80,11 +80,14 @@ def make_default_dtype_config_from_precision(precision: str):
 class McoreConfigsFromPlugin:
     def generate_ddp_config(self):
         return {}
+
     def generate_optim_config(self):
         return {}
+
     def generate_model_config(self):
         return {}
 
+
 class MegatronCustomPrecision(MixedPrecision, McoreConfigsFromPlugin):
     def __init__(
         self,