Skip to content

Commit af4ffe4

Browse files
make style
1 parent 9fde5e4 commit af4ffe4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimum/intel/openvino/quantization.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -681,7 +681,7 @@ def _prepare_causal_lm_dataset(self, quantization_config: OVWeightQuantizationCo
681681
config_dataset = quantization_config.dataset
682682
if isinstance(config_dataset, str):
683683
calibration_dataset = get_dataset(config_dataset, tokenizer, seqlen=32, nsamples=nsamples)
684-
elif isinstance(config_dataset, list) and all([isinstance(it, str) for it in config_dataset]):
684+
elif isinstance(config_dataset, list) and all(isinstance(it, str) for it in config_dataset):
685685
calibration_dataset = [tokenizer(text, return_tensors="pt") for text in config_dataset[:nsamples]]
686686
else:
687687
raise ValueError("Please provide dataset as one of the accepted dataset labels or as a list of strings.")

0 commit comments

Comments
 (0)