[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
NVIDIA · May 9, 2024 · 4ee834c · 4ee834c
1 parent 7d6d511
commit 4ee834c
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/examples/nlp/language_modeling/megatron_retro_eval_legacy.py b/examples/nlp/language_modeling/megatron_retro_eval_legacy.py
@@ -65,7 +65,6 @@
     explanation=f"\n{banner}\nWARNING: This evaluating script is for the native NeMo RETRO model, which will soon be deprecated in future releases. \nThe replacing Mcore-based RETRO model is implemented at nemo/collections/nlp/models/language_modeling/megatron_retro_model.py. \nThe evaluating script for this new model is implemented at examples/nlp/language_modeling/megatron_retro_eval.py."
     f"\nPlease switch to the new script.\n{banner}\n",
 )
-
 @hydra_runner(config_path="conf", config_name="megatron_retro_inference_legacy")
 def main(cfg) -> None:
     trainer = Trainer(strategy=NLPDDPStrategy(), **cfg.trainer)

diff --git a/examples/nlp/language_modeling/megatron_retro_pretraining_legacy.py b/examples/nlp/language_modeling/megatron_retro_pretraining_legacy.py
@@ -36,12 +36,12 @@
 
 banner = '\n'.join(['' "*" * 80] * 5)
 
+
 @deprecated(
     wait_seconds=1,
     explanation=f"\n{banner}\nWARNING: This pretraining script is for the native NeMo RETRO model, which will soon be deprecated in future releases. \nThe replacing Mcore-based RETRO model is implemented at nemo/collections/nlp/models/language_modeling/megatron_retro_model.py. \nThe training script for this new model is implemented at examples/nlp/language_modeling/megatron_retro_pretraining.py."
     f"\nPlease switch to the new script.\n{banner}\n",
 )
-
 @hydra_runner(config_path="conf", config_name="megatron_retro_config_legacy")
 def main(cfg) -> None:
     logging.info("\n\n************** Experiment configuration ***********")