aboutsummaryrefslogtreecommitdiff
path: root/tests/mlia/test_api.py
diff options
context:
space:
mode:
authorBenjamin Klimczak <benjamin.klimczak@arm.com>2022-06-27 09:10:49 +0100
committerBenjamin Klimczak <benjamin.klimczak@arm.com>2022-06-27 16:18:55 +0100
commit0ee201dfa023a9f42e1c8a282e96c231c6769e07 (patch)
treeabd0048d0f1922470e3fe6befcc32f62d38a52f5 /tests/mlia/test_api.py
parent7faf2c4763f299ee53b1ed100025ba50021c8313 (diff)
downloadmlia-0ee201dfa023a9f42e1c8a282e96c231c6769e07.tar.gz
MLIA-522 No 'perf' in mode 'all' for TFLite files
Fix the issue that no performance information is shown for TFLite files when the mode 'all_tests' is used. Change-Id: I8b4df4ab84ba9783b582ad449a34bf6177037e14
Diffstat (limited to 'tests/mlia/test_api.py')
-rw-r--r--tests/mlia/test_api.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/mlia/test_api.py b/tests/mlia/test_api.py
index 54d4796..09bc509 100644
--- a/tests/mlia/test_api.py
+++ b/tests/mlia/test_api.py
@@ -55,7 +55,7 @@ def test_get_advice_wrong_category(test_keras_model: Path) -> None:
[
"all",
ExecutionContext(advice_category=AdviceCategory.PERFORMANCE),
- AdviceCategory.PERFORMANCE,
+ AdviceCategory.ALL,
],
[
"all",