Skip to content

Commit af8960b

Browse files
tests/python_tests/test_vlm_pipeline.py: fix lint issues
1 parent 3d40d8d commit af8960b

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

tests/python_tests/test_vlm_pipeline.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -188,8 +188,12 @@ def _get_ov_model(model_id: str) -> str:
188188
pytest.skip("ValueError: The current version of Transformers does not allow for the export of the model. Maximum required is 4.53.3, got: 4.55.4")
189189
if "optimum-intel-internal-testing/tiny-random-phi3-vision" == model_id:
190190
pytest.xfail("AttributeError: 'DynamicCache' object has no attribute 'get_usable_length'. Ticket CVS-175110")
191-
if "optimum-intel-internal-testing/tiny-random-MiniCPM-o-2_6" == model_id and is_transformers_version(">", "4.51.3"):
192-
pytest.skip("ValueError: The current version of Transformers does not allow for the export of the model. Maximum supported version is 4.51.3")
191+
if "optimum-intel-internal-testing/tiny-random-MiniCPM-o-2_6" == model_id and is_transformers_version(
192+
">", "4.51.3"
193+
):
194+
pytest.skip(
195+
"ValueError: The current version of Transformers does not allow for the export of the model. Maximum supported version is 4.51.3"
196+
)
193197

194198
ov_cache_converted_dir = get_ov_cache_converted_models_dir()
195199
dir_name = str(model_id).replace(os.sep, "_")
@@ -392,10 +396,12 @@ def synthetic_video(pytestconfig):
392396
def synthetic_video_32x32(synthetic_video):
393397
return resize_video(synthetic_video, (32, 32))
394398

399+
395400
@pytest.fixture(scope="module")
396401
def cat_image_448x448(cat_image):
397402
return cat_image.resize((448, 448))
398403

404+
399405
@pytest.fixture(scope="module")
400406
def cat_image_384x384(cat_image):
401407
return cat_image.resize((384, 384))
@@ -1612,7 +1618,9 @@ def get_nanollava_processor():
16121618
# Gemma3 input_ids has two bos tokens when running with optimum: one in chat template + "add_bos_token" is set to True in tokenizer_config.json
16131619
if model.config.model_type == "gemma3":
16141620
processor.tokenizer.add_bos_token = False
1615-
inputs = model.preprocess_inputs(text=prompt, image=resized_image, video=resized_video, processor=processor, config=model.config)
1621+
inputs = model.preprocess_inputs(
1622+
text=prompt, image=resized_image, video=resized_video, processor=processor, config=model.config
1623+
)
16161624

16171625
max_new_tokens = 100
16181626

0 commit comments

Comments
 (0)