From dcd0bd31985c27e1d07333351b26cf8ad12ad1fd Mon Sep 17 00:00:00 2001 From: Benjamin Klimczak Date: Wed, 14 Dec 2022 11:20:11 +0000 Subject: MLIA-589 Create an API to get target information Change-Id: Ieeaa9188ea1e29e2ccaad7475d457bce71e3140d --- tests/test_target_cortex_a_operators.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'tests/test_target_cortex_a_operators.py') diff --git a/tests/test_target_cortex_a_operators.py b/tests/test_target_cortex_a_operators.py index 94eb890..262ebc8 100644 --- a/tests/test_target_cortex_a_operators.py +++ b/tests/test_target_cortex_a_operators.py @@ -6,18 +6,18 @@ from pathlib import Path import pytest import tensorflow as tf +from mlia.backend.armnn_tflite_delegate import compat from mlia.nn.tensorflow.tflite_graph import TFL_OP from mlia.nn.tensorflow.utils import convert_to_tflite -from mlia.target.cortex_a import operator_compatibility as op_compat from mlia.target.cortex_a.operators import CortexACompatibilityInfo from mlia.target.cortex_a.operators import get_cortex_a_compatibility_info from mlia.target.cortex_a.operators import Operator -def test_op_compat_data() -> None: +def test_compat_data() -> None: """Make sure all data contains the necessary items.""" builtin_tfl_ops = {op.name for op in TFL_OP} - for data in [op_compat.ARMNN_TFLITE_DELEGATE]: + for data in [compat.ARMNN_TFLITE_DELEGATE]: assert "metadata" in data assert "backend" in data["metadata"] assert "version" in data["metadata"] -- cgit v1.2.1