blob: 72454bc6533011f1bd6aa51d26588e7468e2d9fc (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
# SPDX-FileCopyrightText: Copyright 2022, 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=["mlia[tosa]"],
packages_to_uninstall=["tosa-checker"],
expected_packages=["tosa-checker"],
)
|