aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/backend/armnn_tflite_delegate/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/mlia/backend/armnn_tflite_delegate/__init__.py')
-rw-r--r--src/mlia/backend/armnn_tflite_delegate/__init__.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/src/mlia/backend/armnn_tflite_delegate/__init__.py b/src/mlia/backend/armnn_tflite_delegate/__init__.py
index 4fe8639..66e5a2a 100644
--- a/src/mlia/backend/armnn_tflite_delegate/__init__.py
+++ b/src/mlia/backend/armnn_tflite_delegate/__init__.py
@@ -1,6 +1,8 @@
# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
"""Arm NN TensorFlow Lite delegate backend module."""
+from typing import cast
+
from mlia.backend.armnn_tflite_delegate.compat import ARMNN_TFLITE_DELEGATE
from mlia.backend.config import BackendConfiguration
from mlia.backend.config import BackendType
@@ -14,5 +16,5 @@ registry.register(
supported_systems=None,
backend_type=BackendType.BUILTIN,
),
- pretty_name=ARMNN_TFLITE_DELEGATE["metadata"]["backend"],
+ pretty_name=cast(str, ARMNN_TFLITE_DELEGATE["backend"]),
)