diff --git a/tests/python_tests/test_llm_pipeline_static.py b/tests/python_tests/test_llm_pipeline_static.py index c9ae3ce30f..b3969fe002 100644 --- a/tests/python_tests/test_llm_pipeline_static.py +++ b/tests/python_tests/test_llm_pipeline_static.py @@ -15,7 +15,7 @@ get_greedy_n_stop_strings_include_to_output, \ get_multinomial_temperature, \ get_multinomial_all_parameters, \ - get_multinomial_temperature_and_presence_penalty \ + get_multinomial_temperature_and_presence_penalty, \ get_beam_search # This test suite is designed specifically to validate the functionality and robustness of the StaticLLMPipeline on NPUW:CPU.