fix tests

Cyrilvallez · Cyrilvallez · commit fce79da0d188 · 2025-11-18T18:52:03.000+01:00
diff --git a/tests/utils/test_core_model_loading.py b/tests/utils/test_core_model_loading.py
@@ -217,7 +217,7 @@ def test_moe_and_qkv_conversion(self):
             ),
             WeightConverter("mlp.w2.weight", "mlp.down_proj.weight"),
         ]
-        missing, unexpected, mismatch, misc = convert_and_load_state_dict_in_model(
+        missing, unexpected, mismatch, _, misc = convert_and_load_state_dict_in_model(
             model, state_dict, weight_mapping, tp_plan=None, hf_quantizer=None
         )
 
@@ -367,7 +367,7 @@ def __init__(self):
             )
         ]
 
-        missing, unexpected, mismatch, misc = convert_and_load_state_dict_in_model(
+        missing, unexpected, mismatch, _, misc = convert_and_load_state_dict_in_model(
             model, state_dict, weight_mapping, tp_plan=None, hf_quantizer=quantizer
         )
 

Original file line number	Diff line number	Diff line change
`@@ -217,7 +217,7 @@ def test_moe_and_qkv_conversion(self):`
`217`	`217`	`),`
`218`	`218`	`WeightConverter("mlp.w2.weight", "mlp.down_proj.weight"),`
`219`	`219`	`]`
`220`		`- missing, unexpected, mismatch, misc = convert_and_load_state_dict_in_model(`
	`220`	`+ missing, unexpected, mismatch, _, misc = convert_and_load_state_dict_in_model(`
`221`	`221`	`model, state_dict, weight_mapping, tp_plan=None, hf_quantizer=None`
`222`	`222`	`)`
`223`	`223`
`@@ -367,7 +367,7 @@ def __init__(self):`
`367`	`367`	`)`
`368`	`368`	`]`
`369`	`369`
`370`		`- missing, unexpected, mismatch, misc = convert_and_load_state_dict_in_model(`
	`370`	`+ missing, unexpected, mismatch, _, misc = convert_and_load_state_dict_in_model(`
`371`	`371`	`model, state_dict, weight_mapping, tp_plan=None, hf_quantizer=quantizer`
`372`	`372`	`)`
`373`	`373`