[Continuous batching] Enable python tests (openvinotoolkit#746)

yatarkan · Aug 6, 2024 · d9324ea · d9324ea
1 parent 50182b4
commit d9324ea
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 6 deletions.
diff --git a/tests/python_tests/test_preemption.py b/tests/python_tests/test_preemption.py
@@ -11,9 +11,6 @@
 from test_sampling import RandomSamplingTestStruct, get_current_plarform_ref_texts
 
 
-pytest.skip("continuous_batching fails with nightly ov", allow_module_level=True)
-
-
 def get_greedy_seq_len_300() -> GenerationConfig:
     generation_config = GenerationConfig()
     generation_config.num_return_sequences = 3

diff --git a/tests/python_tests/test_sampling.py b/tests/python_tests/test_sampling.py
@@ -22,9 +22,6 @@
     generate_and_compare_with_hf, get_multinomial_temperature_and_repetition_penalty, get_scheduler_config
 
 
-pytest.skip("continuous_batching fails with nightly ov", allow_module_level=True)
-
-
 @pytest.mark.precommit
 @pytest.mark.parametrize("model_id", get_models_list(os.path.join(os.path.dirname(os.path.realpath(__file__)), "models", "precommit")))
 @pytest.mark.xfail(