diff options
Diffstat (limited to 'tests/mlia/test_devices_ethosu_advice_generation.py')
| -rw-r--r-- | tests/mlia/test_devices_ethosu_advice_generation.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/tests/mlia/test_devices_ethosu_advice_generation.py b/tests/mlia/test_devices_ethosu_advice_generation.py index 98c8a57..5d37376 100644 --- a/tests/mlia/test_devices_ethosu_advice_generation.py +++ b/tests/mlia/test_devices_ethosu_advice_generation.py @@ -2,7 +2,6 @@ # SPDX-License-Identifier: Apache-2.0 """Tests for Ethos-U advice generation.""" from typing import List -from typing import Optional import pytest @@ -387,7 +386,7 @@ def test_ethosu_advice_producer( "advice_category, action_resolver, expected_advice", [ [ - None, + AdviceCategory.ALL, None, [], ], @@ -467,7 +466,7 @@ def test_ethosu_advice_producer( ) def test_ethosu_static_advice_producer( tmpdir: str, - advice_category: Optional[AdviceCategory], + advice_category: AdviceCategory, action_resolver: ActionResolver, expected_advice: List[Advice], ) -> None: |