diff --git a/tests/benchmarks/micro_benchmarks/test_dist_inference.py b/tests/benchmarks/micro_benchmarks/test_dist_inference.py index 80d3ca2fc..4cf72c9d3 100644 --- a/tests/benchmarks/micro_benchmarks/test_dist_inference.py +++ b/tests/benchmarks/micro_benchmarks/test_dist_inference.py @@ -193,7 +193,7 @@ def _test_dist_inference_result_parsing(self, platform, test_raw_output): (benchmark_class, predefine_params) = BenchmarkRegistry._BenchmarkRegistry__select_benchmark(benchmark_name, platform) assert (benchmark_class) - benchmark = benchmark_class(benchmark_name, parameters='') + benchmark = benchmark_class(benchmark_name, parameters='--num_layers=100') assert (benchmark) ret = benchmark._preprocess() assert (ret is True)