diff options
Diffstat (limited to 'tests/test_devices_cortexa_reporters.py')
-rw-r--r-- | tests/test_devices_cortexa_reporters.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/test_devices_cortexa_reporters.py b/tests/test_devices_cortexa_reporters.py index 4177b55..1110653 100644 --- a/tests/test_devices_cortexa_reporters.py +++ b/tests/test_devices_cortexa_reporters.py @@ -12,6 +12,7 @@ from mlia.devices.cortexa.operators import Operator from mlia.devices.cortexa.reporters import cortex_a_formatters from mlia.devices.cortexa.reporters import report_device from mlia.nn.tensorflow.tflite_compat import TFLiteCompatibilityInfo +from mlia.nn.tensorflow.tflite_compat import TFLiteCompatibilityStatus from mlia.nn.tensorflow.tflite_graph import TFL_ACTIVATION_FUNCTION @@ -25,7 +26,7 @@ def test_report_device() -> None: "data", ( [Advice(["Sample", "Advice"])], - TFLiteCompatibilityInfo(compatible=True), + TFLiteCompatibilityInfo(status=TFLiteCompatibilityStatus.COMPATIBLE), [ Operator( name="Test", |