Fixed UT error

PenghuiCheng · PenghuiCheng · commit ac9aee8643b9 · 2024-03-25T11:26:27.000+08:00
Signed-off-by: Cheng, Penghui &lt;penghui.cheng@intel.com&gt;
diff --git a/tests/neural_compressor/test_optimization.py b/tests/neural_compressor/test_optimization.py
@@ -240,8 +240,9 @@ def test_weight_only_quantization(self, no_config, algo, weight_dtype):
                 )
             else:
                 q_model = quantizer.quantize(
-                    weight_only=True,  # use RTN quantization method and NF4 weight data type is default.
+                    quantization_config=None,
                     save_directory=tmp_dir,
+                    weight_only=True,  # use RTN quantization method and NF4 weight data type is default.
                 )
             q_model = INCModelForCausalLM.from_pretrained(tmp_dir)
             inp = torch.tensor([calibration_dataset[0]["input_ids"]])

Original file line number	Diff line number	Diff line change
`@@ -240,8 +240,9 @@ def test_weight_only_quantization(self, no_config, algo, weight_dtype):`
`240`	`240`	`)`
`241`	`241`	`else:`
`242`	`242`	`q_model = quantizer.quantize(`
`243`		`- weight_only=True, # use RTN quantization method and NF4 weight data type is default.`
	`243`	`+ quantization_config=None,`
`244`	`244`	`save_directory=tmp_dir,`
	`245`	`+ weight_only=True, # use RTN quantization method and NF4 weight data type is default.`
`245`	`246`	`)`
`246`	`247`	`q_model = INCModelForCausalLM.from_pretrained(tmp_dir)`
`247`	`248`	`inp = torch.tensor([calibration_dataset[0]["input_ids"]])`