aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/backend
diff options
context:
space:
mode:
authorDmitrii Agibov <dmitrii.agibov@arm.com>2022-12-13 09:39:21 +0000
committerDmitrii Agibov <dmitrii.agibov@arm.com>2022-12-16 09:25:21 +0000
commit8da6441c87532d20d3c8dad538e46dc99d073927 (patch)
tree6afb7f0c82c5c357f7497413f0e18473e84d472c /src/mlia/backend
parentba9aeace4c3d3f4f31830d68b5cc7dd3f4bf1dde (diff)
downloadmlia-8da6441c87532d20d3c8dad538e46dc99d073927.tar.gz
MLIA-460 Revisit backend management
- Provide command for backend installation in case if backend is not available - Fix issue with connection timeout during downloading - Show installation tools output only in verbose mode Change-Id: Ic0e495ba19879cc2cda4fd0bce20b57ba896cfeb
Diffstat (limited to 'src/mlia/backend')
-rw-r--r--src/mlia/backend/errors.py12
-rw-r--r--src/mlia/backend/tosa_checker/compat.py6
2 files changed, 15 insertions, 3 deletions
diff --git a/src/mlia/backend/errors.py b/src/mlia/backend/errors.py
new file mode 100644
index 0000000..bd5da95
--- /dev/null
+++ b/src/mlia/backend/errors.py
@@ -0,0 +1,12 @@
+# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+"""Backend errors."""
+
+
+class BackendUnavailableError(Exception):
+ """Backend unavailable error."""
+
+ def __init__(self, msg: str, backend: str) -> None:
+ """Init error."""
+ super().__init__(msg)
+ self.backend = backend
diff --git a/src/mlia/backend/tosa_checker/compat.py b/src/mlia/backend/tosa_checker/compat.py
index e1bcb24..bd21774 100644
--- a/src/mlia/backend/tosa_checker/compat.py
+++ b/src/mlia/backend/tosa_checker/compat.py
@@ -8,6 +8,7 @@ from typing import Any
from typing import cast
from typing import Protocol
+from mlia.backend.errors import BackendUnavailableError
from mlia.core.typing import PathOrFileLike
@@ -45,9 +46,8 @@ def get_tosa_compatibility_info(
checker = get_tosa_checker(tflite_model_path)
if checker is None:
- raise Exception(
- "TOSA checker is not available. "
- "Please make sure that 'tosa-checker' backend is installed."
+ raise BackendUnavailableError(
+ "Backend tosa-checker is not available", "tosa-checker"
)
ops = [