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

for more information, see https://pre-commit.ci
NVIDIA · Sep 26, 2023 · 772f93b · 772f93b
1 parent 19b831f
commit 772f93b
Show file tree

Hide file tree

Showing 14 changed files with 70 additions and 28 deletions.
diff --git a/examples/nlp/language_modeling/megatron_t5_prompt_learning.py b/examples/nlp/language_modeling/megatron_t5_prompt_learning.py
@@ -43,8 +43,11 @@
 within this examples directory to convert your model to .nemo format.
 """
 
-@deprecated(explanation=f"{__file__} is deprecated. Please use MegatronT5SFTModel.add_adapter() for PEFT features."
-            "See updated scripts `megatron_t5_peft_tuning.py` and `megatron_t5_peft_eval.py` for examples.")
+
+@deprecated(
+    explanation=f"{__file__} is deprecated. Please use MegatronT5SFTModel.add_adapter() for PEFT features."
+    "See updated scripts `megatron_t5_peft_tuning.py` and `megatron_t5_peft_eval.py` for examples."
+)
 @hydra_runner(config_path="conf", config_name="megatron_t5_prompt_learning.yaml")
 def main(cfg) -> None:
     logging.info("\n\n************** Experiment configuration ***********")

diff --git a/examples/nlp/language_modeling/megatron_t5_prompt_learning_eval.py b/examples/nlp/language_modeling/megatron_t5_prompt_learning_eval.py
@@ -37,8 +37,11 @@
 if not torch.cuda.is_available():
     raise EnvironmentError("GPU is needed for the inference")
 
-@deprecated(explanation=f"{__file__} is deprecated. Please use MegatronT5SFTModel.add_adapter() for PEFT features."
-            "See updated scripts `megatron_t5_peft_tuning.py` and `megatron_t5_peft_eval.py` for examples.")
+
+@deprecated(
+    explanation=f"{__file__} is deprecated. Please use MegatronT5SFTModel.add_adapter() for PEFT features."
+    "See updated scripts `megatron_t5_peft_tuning.py` and `megatron_t5_peft_eval.py` for examples."
+)
 @hydra_runner(config_path="conf", config_name="megatron_t5_prompt_learning_inference")
 def main(cfg) -> None:
 

diff --git a/examples/nlp/language_modeling/tuning/megatron_gpt_adapter_eval.py b/examples/nlp/language_modeling/tuning/megatron_gpt_adapter_eval.py
@@ -47,8 +47,11 @@
 if not torch.cuda.is_available():
     raise EnvironmentError("GPU is needed for the inference")
 
-@deprecated(explanation=f"{__file__} is deprecated. Please use MegatronGPTSFTModel.add_adapter() for PEFT features."
-            "See updated scripts `megatron_gpt_peft_tuning.py` and `megatron_gpt_peft_eval.py` for examples.")
+
+@deprecated(
+    explanation=f"{__file__} is deprecated. Please use MegatronGPTSFTModel.add_adapter() for PEFT features."
+    "See updated scripts `megatron_gpt_peft_tuning.py` and `megatron_gpt_peft_eval.py` for examples."
+)
 @hydra_runner(config_path="conf", config_name="megatron_gpt_adapter_inference")
 def main(cfg) -> None:
 

diff --git a/examples/nlp/language_modeling/tuning/megatron_gpt_adapter_tuning.py b/examples/nlp/language_modeling/tuning/megatron_gpt_adapter_tuning.py
@@ -56,8 +56,11 @@
             trainer.max_epochs=2
 """
 
-@deprecated(explanation=f"{__file__} is deprecated. Please use MegatronGPTSFTModel.add_adapter() for PEFT features."
-            "See updated scripts `megatron_gpt_peft_tuning.py` and `megatron_gpt_peft_eval.py` for examples.")
+
+@deprecated(
+    explanation=f"{__file__} is deprecated. Please use MegatronGPTSFTModel.add_adapter() for PEFT features."
+    "See updated scripts `megatron_gpt_peft_tuning.py` and `megatron_gpt_peft_eval.py` for examples."
+)
 @hydra_runner(config_path="conf", config_name="megatron_gpt_adapter_tuning_config")
 def main(cfg) -> None:
     logging.info("\n\n************** Experiment configuration ***********")

diff --git a/examples/nlp/language_modeling/tuning/megatron_gpt_ia3_eval.py b/examples/nlp/language_modeling/tuning/megatron_gpt_ia3_eval.py
@@ -46,8 +46,11 @@
 if not torch.cuda.is_available():
     raise EnvironmentError("GPU is needed for the inference")
 
-@deprecated(explanation=f"{__file__} is deprecated. Please use MegatronGPTSFTModel.add_adapter() for PEFT features."
-            "See updated scripts `megatron_gpt_peft_tuning.py` and `megatron_gpt_peft_eval.py` for examples.")
+
+@deprecated(
+    explanation=f"{__file__} is deprecated. Please use MegatronGPTSFTModel.add_adapter() for PEFT features."
+    "See updated scripts `megatron_gpt_peft_tuning.py` and `megatron_gpt_peft_eval.py` for examples."
+)
 @hydra_runner(config_path="conf", config_name="megatron_gpt_adapter_inference")
 def main(cfg) -> None:
 

diff --git a/examples/nlp/language_modeling/tuning/megatron_gpt_ia3_tuning.py b/examples/nlp/language_modeling/tuning/megatron_gpt_ia3_tuning.py
@@ -56,8 +56,11 @@
             trainer.max_epochs=2
 """
 
-@deprecated(explanation=f"{__file__} is deprecated. Please use MegatronGPTSFTModel.add_adapter() for PEFT features."
-            "See updated scripts `megatron_gpt_peft_tuning.py` and `megatron_gpt_peft_eval.py` for examples.")
+
+@deprecated(
+    explanation=f"{__file__} is deprecated. Please use MegatronGPTSFTModel.add_adapter() for PEFT features."
+    "See updated scripts `megatron_gpt_peft_tuning.py` and `megatron_gpt_peft_eval.py` for examples."
+)
 @hydra_runner(config_path="conf", config_name="megatron_gpt_ia3_tuning_config")
 def main(cfg) -> None:
     logging.info("\n\n************** Experiment configuration ***********")

diff --git a/examples/nlp/language_modeling/tuning/megatron_gpt_sft.py b/examples/nlp/language_modeling/tuning/megatron_gpt_sft.py
@@ -142,8 +142,11 @@ def validate_checkpoint_loading_args(cfg):
     if cfg.hparams_file is None or not os.path.isfile(cfg.hparams_file):
         raise ValueError(f'Hparams file {cfg.hparams_file} does not exist or is not a file.')
 
-@deprecated(explanation=f"{__file__} is deprecated. PEFT and SFT scripts are now consolidated"
-            "See updated scripts `megatron_gpt_peft_tuning.py` and `megatron_gpt_peft_eval.py` for examples.")
+
+@deprecated(
+    explanation=f"{__file__} is deprecated. PEFT and SFT scripts are now consolidated"
+    "See updated scripts `megatron_gpt_peft_tuning.py` and `megatron_gpt_peft_eval.py` for examples."
+)
 @hydra_runner(config_path="conf", config_name="megatron_gpt_sft")
 def main(cfg) -> None:
     logging.info("\n\n************** Experiment configuration ***********")

diff --git a/examples/nlp/language_modeling/tuning/megatron_t5_adapter_eval.py b/examples/nlp/language_modeling/tuning/megatron_t5_adapter_eval.py
@@ -45,8 +45,11 @@
 if not torch.cuda.is_available():
     raise EnvironmentError("GPU is needed for the inference")
 
-@deprecated(explanation=f"{__file__} is deprecated. Please use MegatronT5SFTModel.add_adapter() for PEFT features."
-            "See updated scripts `megatron_t5_peft_tuning.py` and `megatron_t5_peft_eval.py` for examples.")
+
+@deprecated(
+    explanation=f"{__file__} is deprecated. Please use MegatronT5SFTModel.add_adapter() for PEFT features."
+    "See updated scripts `megatron_t5_peft_tuning.py` and `megatron_t5_peft_eval.py` for examples."
+)
 @hydra_runner(config_path="conf", config_name="megatron_t5_adapter_inference")
 def main(cfg) -> None:
 

diff --git a/examples/nlp/language_modeling/tuning/megatron_t5_adapter_tuning.py b/examples/nlp/language_modeling/tuning/megatron_t5_adapter_tuning.py
@@ -56,8 +56,11 @@
             trainer.max_epochs=2
 """
 
-@deprecated(explanation=f"{__file__} is deprecated. Please use MegatronT5SFTModel.add_adapter() for PEFT features."
-            "See updated scripts `megatron_t5_peft_tuning.py` and `megatron_t5_peft_eval.py` for examples.")
+
+@deprecated(
+    explanation=f"{__file__} is deprecated. Please use MegatronT5SFTModel.add_adapter() for PEFT features."
+    "See updated scripts `megatron_t5_peft_tuning.py` and `megatron_t5_peft_eval.py` for examples."
+)
 @hydra_runner(config_path="conf", config_name="megatron_t5_adapter_tuning_config")
 def main(cfg) -> None:
     logging.info("\n\n************** Experiment configuration ***********")

diff --git a/examples/nlp/language_modeling/tuning/megatron_t5_ia3_eval.py b/examples/nlp/language_modeling/tuning/megatron_t5_ia3_eval.py
@@ -45,8 +45,11 @@
 if not torch.cuda.is_available():
     raise EnvironmentError("GPU is needed for the inference")
 
-@deprecated(explanation=f"{__file__} is deprecated. Please use MegatronT5SFTModel.add_adapter() for PEFT features."
-            "See updated scripts `megatron_t5_peft_tuning.py` and `megatron_t5_peft_eval.py` for examples.")
+
+@deprecated(
+    explanation=f"{__file__} is deprecated. Please use MegatronT5SFTModel.add_adapter() for PEFT features."
+    "See updated scripts `megatron_t5_peft_tuning.py` and `megatron_t5_peft_eval.py` for examples."
+)
 @hydra_runner(config_path="conf", config_name="megatron_t5_ia3_inference")
 def main(cfg) -> None:
 

diff --git a/examples/nlp/language_modeling/tuning/megatron_t5_ia3_tuning.py b/examples/nlp/language_modeling/tuning/megatron_t5_ia3_tuning.py
@@ -56,8 +56,11 @@
             trainer.max_epochs=2
 """
 
-@deprecated(explanation=f"{__file__} is deprecated. Please use MegatronT5SFTModel.add_adapter() for PEFT features."
-            "See updated scripts `megatron_t5_peft_tuning.py` and `megatron_t5_peft_eval.py` for examples.")
+
+@deprecated(
+    explanation=f"{__file__} is deprecated. Please use MegatronT5SFTModel.add_adapter() for PEFT features."
+    "See updated scripts `megatron_t5_peft_tuning.py` and `megatron_t5_peft_eval.py` for examples."
+)
 @hydra_runner(config_path="conf", config_name="megatron_t5_ia3_tuning_config")
 def main(cfg) -> None:
     logging.info("\n\n************** Experiment configuration ***********")

diff --git a/examples/nlp/language_modeling/tuning/megatron_t5_lora_eval.py b/examples/nlp/language_modeling/tuning/megatron_t5_lora_eval.py
@@ -45,8 +45,11 @@
 if not torch.cuda.is_available():
     raise EnvironmentError("GPU is needed for the inference")
 
-@deprecated(explanation=f"{__file__} is deprecated. Please use MegatronT5SFTModel.add_adapter() for PEFT features."
-            "See updated scripts `megatron_t5_peft_tuning.py` and `megatron_t5_peft_eval.py` for examples.")
+
+@deprecated(
+    explanation=f"{__file__} is deprecated. Please use MegatronT5SFTModel.add_adapter() for PEFT features."
+    "See updated scripts `megatron_t5_peft_tuning.py` and `megatron_t5_peft_eval.py` for examples."
+)
 @hydra_runner(config_path="conf", config_name="megatron_t5_adapter_inference")
 def main(cfg) -> None:
 

diff --git a/examples/nlp/language_modeling/tuning/megatron_t5_lora_tuning.py b/examples/nlp/language_modeling/tuning/megatron_t5_lora_tuning.py
@@ -56,8 +56,11 @@
             trainer.max_epochs=2
 """
 
-@deprecated(explanation=f"{__file__} is deprecated. Please use MegatronT5SFTModel.add_adapter() for PEFT features."
-            "See updated scripts `megatron_t5_peft_tuning.py` and `megatron_t5_peft_eval.py` for examples.")
+
+@deprecated(
+    explanation=f"{__file__} is deprecated. Please use MegatronT5SFTModel.add_adapter() for PEFT features."
+    "See updated scripts `megatron_t5_peft_tuning.py` and `megatron_t5_peft_eval.py` for examples."
+)
 @hydra_runner(config_path="conf", config_name="megatron_t5_lora_tuning_config")
 def main(cfg) -> None:
     logging.info("\n\n************** Experiment configuration ***********")

diff --git a/nemo/collections/nlp/models/language_modeling/megatron_gpt_peft_models.py b/nemo/collections/nlp/models/language_modeling/megatron_gpt_peft_models.py
@@ -46,8 +46,11 @@
 
     HAVE_MEGATRON_CORE = False
 
-@deprecated(explanation="Please use MegatronGPTSFTModel.add_adapter() for PEFT features."
-                        "See the updated `megatron_gpt_peft_tuning.py` for an example.")
+
+@deprecated(
+    explanation="Please use MegatronGPTSFTModel.add_adapter() for PEFT features."
+    "See the updated `megatron_gpt_peft_tuning.py` for an example."
+)
 class MegatronGPTPEFTModel(MegatronGPTSFTModel):
     """
     base class for all mixin based adapter models