Apply isort and black reformatting

Signed-off-by: akoumpa <[email protected]>
NVIDIA · Sep 19, 2024 · ab0e66a · ab0e66a
1 parent 6a238b9
commit ab0e66a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/tests/collections/llm/auto_conf/test_base_configs.py b/tests/collections/llm/auto_conf/test_base_configs.py
@@ -1,6 +1,4 @@
-from nemo.collections.llm.utils import Config, Partial
 import torch
-
 from megatron.core.optimizer import OptimizerConfig
 from pytorch_lightning.loggers import TensorBoardLogger
 
@@ -17,6 +15,7 @@
 )
 from nemo.collections.llm.tools.auto_configurator import AutoConfigurator
 from nemo.collections.llm.tools.auto_configurator.core.base_config import BaseConfig
+from nemo.collections.llm.utils import Config, Partial
 from nemo.lightning.pytorch.optim import CosineAnnealingScheduler, MegatronOptimizerModule
 from nemo.utils.exp_manager import TimingCallback
 

diff --git a/tests/collections/llm/auto_conf/test_generate_configs.py b/tests/collections/llm/auto_conf/test_generate_configs.py
@@ -1,5 +1,3 @@
-from nemo.collections.llm.utils import Config, Partial
-
 from nemo.collections.llm import (
     GemmaConfig7B,
     GPTConfig5B,
@@ -9,6 +7,7 @@
     Nemotron3Config8B,
 )
 from nemo.collections.llm.tools.auto_configurator import AutoConfigurator, generate_configs
+from nemo.collections.llm.utils import Config, Partial
 
 
 def get_auto_configs(configs):