fix more tests

Author: eaidova

optimum/intel/openvino/quantization.py

@@ -872,7 +872,10 @@ def _prepare_speech_to_text_calibration_data(self, config: OVQuantizationConfigB
             if decoder_w_p_model is not None:
                 decoder_w_p_model.request = decoder_w_p_model.request.request
 
-        datasets = [nncf.Dataset(encoder_calibration_data), nncf.Dataset(decoder_calibration_data),]
+        datasets = [
+            nncf.Dataset(encoder_calibration_data),
+            nncf.Dataset(decoder_calibration_data),
+        ]
         if decoder_w_p_model is not None:
             datasets.append(nncf.Dataset(decoder_w_p_calibration_data))
         return datasets

tests/openvino/test_modeling.py

@@ -535,8 +535,9 @@ def test_seq2seq_load_from_hub(self):
         with TemporaryDirectory() as tmpdirname:
             ov_exported_pipe.save_pretrained(tmpdirname)
             folder_contents = os.listdir(tmpdirname)
-            self.assertTrue(OV_DECODER_WITH_PAST_NAME in folder_contents)
-            self.assertTrue(OV_DECODER_WITH_PAST_NAME.replace(".xml", ".bin") in folder_contents)
+            if not ov_exported_pipe.model.decoder.stateful:
+                self.assertTrue(OV_DECODER_WITH_PAST_NAME in folder_contents)
+                self.assertTrue(OV_DECODER_WITH_PAST_NAME.replace(".xml", ".bin") in folder_contents)
             ov_exported_pipe = optimum_pipeline("text2text-generation", tmpdirname, accelerator="openvino")
             self.assertIsInstance(ov_exported_pipe.model, OVBaseModel)
 

tests/openvino/test_quantization.py

@@ -1230,7 +1230,7 @@ def test_calibration_data_uniqueness(self, model_name, apply_caching):
 
         for inputs_dict in calibration_data:
             for k, v in inputs_dict.items():
-                if k == "input_ids":
+                if k in ["input_ids", "beam_idx"]:
                     continue
 
                 x = (v.numpy() if isinstance(v, torch.Tensor) else v).copy()