diff options
Diffstat (limited to 'tests/test_target_tosa_advisor.py')
-rw-r--r-- | tests/test_target_tosa_advisor.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/test_target_tosa_advisor.py b/tests/test_target_tosa_advisor.py index 36e52e9..cc47321 100644 --- a/tests/test_target_tosa_advisor.py +++ b/tests/test_target_tosa_advisor.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates. +# SPDX-FileCopyrightText: Copyright 2022-2024, Arm Limited and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 """Tests for TOSA advisor.""" from pathlib import Path @@ -46,7 +46,8 @@ def test_configure_and_get_tosa_advisor( "optimization_type": "clustering", }, ] - ] + ], + "training_parameters": [None], }, "tosa_inference_advisor": { "model": str(test_tflite_model), |