From 37959522a805a5e23c930ed79aac84920c3cb208 Mon Sep 17 00:00:00 2001 From: Dmitrii Agibov Date: Fri, 18 Nov 2022 16:34:03 +0000 Subject: Move backends functionality into separate modules - Move backend management/executor code into module backend_core - Create separate module for each backend in "backend" module - Move each backend into corresponding module - Split Vela wrapper into several submodules Change-Id: If01b6774aab6501951212541cc5d7f5aa7c97e95 --- src/mlia/cli/config.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) (limited to 'src/mlia/cli') diff --git a/src/mlia/cli/config.py b/src/mlia/cli/config.py index 6ea9bb4..2d694dc 100644 --- a/src/mlia/cli/config.py +++ b/src/mlia/cli/config.py @@ -6,18 +6,19 @@ from __future__ import annotations import logging from functools import lru_cache -import mlia.backend.manager as backend_manager -from mlia.tools.metadata.common import DefaultInstallationManager -from mlia.tools.metadata.common import InstallationManager -from mlia.tools.metadata.corstone import get_corstone_installations -from mlia.tools.metadata.py_package import get_pypackage_backend_installations +from mlia.backend.corstone.install import get_corstone_installations +from mlia.backend.install import supported_backends +from mlia.backend.manager import DefaultInstallationManager +from mlia.backend.manager import InstallationManager +from mlia.backend.tosa_checker.install import get_tosa_backend_installation logger = logging.getLogger(__name__) def get_installation_manager(noninteractive: bool = False) -> InstallationManager: """Return installation manager.""" - backends = get_corstone_installations() + get_pypackage_backend_installations() + backends = get_corstone_installations() + backends.append(get_tosa_backend_installation()) return DefaultInstallationManager(backends, noninteractive=noninteractive) @@ -31,7 +32,7 @@ def get_available_backends() -> list[str]: manager = get_installation_manager() available_backends.extend( backend - for backend in backend_manager.supported_backends() + for backend in supported_backends() if manager.backend_installed(backend) ) -- cgit v1.2.1