From 164505e17907f362963b958cfb0fe003e2177dbe Mon Sep 17 00:00:00 2001 From: Alexander Suvorov Date: Mon, 22 Jul 2024 11:39:52 +0200 Subject: [PATCH] Unskip ov_tensor tests --- tests/python_tests/test_generate_api.py | 36 ++++++++++++------------- 1 file changed, 18 insertions(+), 18 deletions(-) diff --git a/tests/python_tests/test_generate_api.py b/tests/python_tests/test_generate_api.py index 40bc121293..039f63c314 100644 --- a/tests/python_tests/test_generate_api.py +++ b/tests/python_tests/test_generate_api.py @@ -161,12 +161,12 @@ def test_decoding(model_descr, generation_config, prompt): ] @pytest.mark.parametrize("inputs", input_tensors_list) @pytest.mark.parametrize("model_descr", get_models_list()) -@pytest.mark.xfail( - raises=TypeError, - reason="pybind was unable to find ov::Tensor from openvino yet", - strict=False, - condition=sys.platform in ["linux", "win32"] -) +# @pytest.mark.xfail( +# raises=TypeError, +# reason="pybind was unable to find ov::Tensor from openvino yet", +# strict=False, +# condition=sys.platform in ["linux", "win32"] +# ) @pytest.mark.precommit def test_ov_tensors(model_descr, inputs): hf_ov_genai_tensors_comparison(read_model(model_descr), dict(max_new_tokens=20), *inputs) @@ -182,12 +182,12 @@ def test_ov_tensors(model_descr, inputs): @pytest.mark.parametrize("model_descr", get_models_list()) @pytest.mark.parametrize("prompt", prompts) @pytest.mark.precommit -@pytest.mark.xfail( - raises=TypeError, - reason="pybind was unable to find ov::Tensor from openvino yet", - strict=False, - condition=sys.platform in ["linux", "win32"] -) +# @pytest.mark.xfail( +# raises=TypeError, +# reason="pybind was unable to find ov::Tensor from openvino yet", +# strict=False, +# condition=sys.platform in ["linux", "win32"] +# ) def test_genai_tokenizer_encode(model_descr, prompt): model_id, path, tokenizer, model, pipe = read_model(model_descr) tok = pipe.get_tokenizer() @@ -217,12 +217,12 @@ def test_genai_tokenizer_encode(model_descr, prompt): @pytest.mark.parametrize("model_descr", get_models_list()) @pytest.mark.parametrize("encoded_prompt", encoded_prompts) @pytest.mark.precommit -@pytest.mark.xfail( - raises=TypeError, - reason="pybind was unable to find ov::Tensor from openvino yet", - strict=False, - condition=sys.platform in ["linux", "win32"] -) +# @pytest.mark.xfail( +# raises=TypeError, +# reason="pybind was unable to find ov::Tensor from openvino yet", +# strict=False, +# condition=sys.platform in ["linux", "win32"] +# ) def test_genai_tokenizer_decode(model_descr, encoded_prompt): model_id, path, tokenizer, model, pipe = read_model(model_descr) tok = pipe.get_tokenizer()