fix: apply ruff formatting to mcore_qwen3vl plugin and test files

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
Signed-off-by: Hung-Yueh Chiang <hungyuehc@nvidia.com>

Author: hychiang-git

modelopt/torch/export/plugins/mcore_common.py

@@ -39,10 +39,7 @@
     qwen25_causal_lm_export,
     qwen25_causal_lm_import,
 )
-from .mcore_qwen3vl import (
-    qwen3vl_causal_lm_export,
-    qwen3vl_causal_lm_import,
-)
+from .mcore_qwen3vl import qwen3vl_causal_lm_export, qwen3vl_causal_lm_import
 
 all_mcore_hf_export_mapping: dict[str, Any] = {
     "DeepseekV2ForCausalLM": deepseek_causal_lm_export,

modelopt/torch/export/plugins/mcore_qwen3vl.py

@@ -117,4 +117,4 @@
     "local_experts.linear_fc2": NameRemapping(
         "model.language_model.layers.{}.mlp.experts.{}.down_proj."
     ),
-}
+}

tests/gpu_megatron/torch/export/test_mcore_qwen3vl.py

@@ -121,9 +121,7 @@ def test_qkv_uses_merging(self):
         assert isinstance(qwen3vl_causal_lm_import["linear_qkv"], QKVMerging)
 
     def test_mlp_uses_gated_merging(self):
-        assert isinstance(
-            qwen3vl_causal_lm_import["linear_fc1"], GatedMLPMerging
-        )
+        assert isinstance(qwen3vl_causal_lm_import["linear_fc1"], GatedMLPMerging)
 
     @pytest.mark.parametrize(
         "key",
@@ -192,14 +190,17 @@ def test_qkv_uses_slicing(self):
         assert isinstance(qwen3vl_causal_lm_export["linear_qkv"], QKVSlicing)
 
     def test_mlp_uses_gated_slicing(self):
-        assert isinstance(
-            qwen3vl_causal_lm_export["linear_fc1"], GatedMLPSlicing
-        )
+        assert isinstance(qwen3vl_causal_lm_export["linear_fc1"], GatedMLPSlicing)
 
     def test_export_has_no_parallel_config(self):
         """Export mappings should not have parallel configs."""
-        for key in ["word_embeddings", "final_layernorm", "output_layer",
-                     "input_layernorm", "linear_proj"]:
+        for key in [
+            "word_embeddings",
+            "final_layernorm",
+            "output_layer",
+            "input_layernorm",
+            "linear_proj",
+        ]:
             mapping = qwen3vl_causal_lm_export[key]
             assert "parallel_config" not in mapping.func_kwargs
 
@@ -208,9 +209,7 @@ class TestQwen3VLImportExportSymmetry:
     """Test that import and export mappings are consistent."""
 
     def test_same_mcore_keys(self):
-        assert set(qwen3vl_causal_lm_import.keys()) == set(
-            qwen3vl_causal_lm_export.keys()
-        )
+        assert set(qwen3vl_causal_lm_import.keys()) == set(qwen3vl_causal_lm_export.keys())
 
     @pytest.mark.parametrize(
         "key",
@@ -256,12 +255,8 @@ def test_same_keys_as_qwen3(self):
             qwen3_causal_lm_import,
         )
 
-        assert set(qwen3vl_causal_lm_import.keys()) == set(
-            qwen3_causal_lm_import.keys()
-        )
-        assert set(qwen3vl_causal_lm_export.keys()) == set(
-            qwen3_causal_lm_export.keys()
-        )
+        assert set(qwen3vl_causal_lm_import.keys()) == set(qwen3_causal_lm_import.keys())
+        assert set(qwen3vl_causal_lm_export.keys()) == set(qwen3_causal_lm_export.keys())
 
     @pytest.mark.parametrize(
         "key",
@@ -290,9 +285,7 @@ def test_vl_adds_language_model_prefix(self, key):
         qwen3_prefix = qwen3_causal_lm_import[key].target_name_or_prefix
         qwen3vl_prefix = qwen3vl_causal_lm_import[key].target_name_or_prefix
         expected = qwen3_prefix.replace("model.", "model.language_model.", 1)
-        assert qwen3vl_prefix == expected, (
-            f"{key}: expected '{expected}', got '{qwen3vl_prefix}'"
-        )
+        assert qwen3vl_prefix == expected, f"{key}: expected '{expected}', got '{qwen3vl_prefix}'"
 
     def test_output_layer_same(self):
         """lm_head is at root level for both Qwen3 and Qwen3-VL."""