ruff format

Signed-off-by: Peter St. John <pstjohn@nvidia.com>

Author: pstjohn

models/.ruff.toml

@@ -2,7 +2,7 @@ line-length = 119
 target-version = "py312"
 
 [lint]
-ignore = ["D100", "E501", "N811", "N814"]
+ignore = ["C901", "D100", "E501", "N811", "N814"]
 select = [
     "C",    # Pylint conventions
     "D",    # Documentation formatting

models/esm2/tests/test_export.py

@@ -15,9 +15,10 @@
 
 
 def test_export_hf_checkpoint(tmp_path):
-    from esm.export import export_hf_checkpoint
     from transformers import AutoModel, AutoModelForMaskedLM, AutoTokenizer
 
+    from esm.export import export_hf_checkpoint
+
     export_hf_checkpoint("esm2_t6_8M_UR50D", tmp_path)
 
     model_for_masked_lm, loading_info = AutoModelForMaskedLM.from_pretrained(

models/esm2/tests/test_modeling_esm_te.py

@@ -169,9 +169,10 @@ def test_golden_values(input_data):
 
 
 def test_converted_model_roundtrip(tmp_path, input_data):
+    from transformer_engine.pytorch import TransformerLayer
+
     from esm.convert import convert_esm_hf_to_te
     from esm.modeling_esm_te import NVEsmConfig, NVEsmEncoder, NVEsmForMaskedLM, NVEsmLMHead, NVEsmModel
-    from transformer_engine.pytorch import TransformerLayer
 
     model_hf = AutoModelForMaskedLM.from_pretrained("facebook/esm2_t6_8M_UR50D")
     model_te = convert_esm_hf_to_te(model_hf)