blob: dd2f0075509b25b960a5b192553e344932c2f870 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
"""Various util functions."""
from importlib import metadata
from pathlib import Path
from mlia.utils.filesystem import sha256
class MetadataError(Exception):
"""Metadata error."""
def yes(prompt: str) -> bool:
"""Return true if user confirms the action."""
response = input(f"{prompt} [y/n]: ")
return response in ["y", "Y"]
def get_pkg_version(pkg_name: str) -> str:
"""Return the version of python package."""
try:
pkg_version = metadata.version(pkg_name)
except FileNotFoundError as exc:
raise MetadataError(exc) from exc
return pkg_version
def get_file_checksum(input_path: Path) -> str:
"""Retrun the checksum of the input model."""
return sha256(input_path)
|
