diff options
Diffstat (limited to 'src/mlia/backend/tosa_checker/install.py')
-rw-r--r-- | src/mlia/backend/tosa_checker/install.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/mlia/backend/tosa_checker/install.py b/src/mlia/backend/tosa_checker/install.py index 72454bc..ba582ed 100644 --- a/src/mlia/backend/tosa_checker/install.py +++ b/src/mlia/backend/tosa_checker/install.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates. +# 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 @@ -13,7 +13,7 @@ def get_tosa_backend_installation() -> Installation: name="tosa-checker", description="Tool to check if a ML model is compatible " "with the TOSA specification", - packages_to_install=["mlia[tosa]"], + packages_to_install=["tosa-checker"], packages_to_uninstall=["tosa-checker"], expected_packages=["tosa-checker"], ) |