blob: ba582ed954ad1e5be1150daa1c468aa35a5ef791 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
"""Module for python package based installations."""
from __future__ import annotations
from mlia.backend.install import Installation
from mlia.backend.install import PyPackageBackendInstallation
def get_tosa_backend_installation() -> Installation:
"""Get TOSA backend installation."""
return PyPackageBackendInstallation(
name="tosa-checker",
description="Tool to check if a ML model is compatible "
"with the TOSA specification",
packages_to_install=["tosa-checker"],
packages_to_uninstall=["tosa-checker"],
expected_packages=["tosa-checker"],
)
|