Apply isort and black reformatting

Signed-off-by: akoumpa <akoumpa@users.noreply.github.com>
NVIDIA · Aug 13, 2024 · 715a3db · 715a3db
1 parent 9140e4b
commit 715a3db
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git 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
@@ -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
@@ -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,