Fix test

nikita-savelyevv · nikita-savelyevv · commit 0b0d54c26bf6 · 2025-03-06T16:51:17.000+01:00
diff --git a/optimum/intel/openvino/quantization/quantizer.py b/optimum/intel/openvino/quantization/quantizer.py
@@ -203,7 +203,7 @@ def quantize(
                 quantization_config.dataset = calibration_dataset
 
             calibration_dataset = self.dataset_builder.build_from_dataset(
-                quantization_config, calibration_dataset, batch_size, remove_unused_columns
+                quantization_config, calibration_dataset, batch_size, data_collator, remove_unused_columns
             )
 
         from ..modeling_base import OVBaseModel

Original file line number	Diff line number	Diff line change
`@@ -203,7 +203,7 @@ def quantize(`
`203`	`203`	`quantization_config.dataset = calibration_dataset`
`204`	`204`
`205`	`205`	`calibration_dataset = self.dataset_builder.build_from_dataset(`
`206`		`- quantization_config, calibration_dataset, batch_size, remove_unused_columns`
	`206`	`+ quantization_config, calibration_dataset, batch_size, data_collator, remove_unused_columns`
`207`	`207`	`)`
`208`	`208`
`209`	`209`	`from ..modeling_base import OVBaseModel`