Skip to content

Commit

Permalink
Unskip ov_tensor tests
Browse files Browse the repository at this point in the history
  • Loading branch information
as-suvorov committed Jul 22, 2024
1 parent 0c2b68e commit 164505e
Showing 1 changed file with 18 additions and 18 deletions.
36 changes: 18 additions & 18 deletions tests/python_tests/test_generate_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand All @@ -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()
Expand Down Expand Up @@ -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()
Expand Down

0 comments on commit 164505e

Please sign in to comment.