1 files changed, 23 insertions, 1 deletions
diff --git a/src/mlia/utils/misc.py b/src/mlia/utils/misc.py
index de95448..dd2f007 100644
--- a/src/mlia/utils/misc.py
+++ b/src/mlia/utils/misc.py
@@ -1,9 +1,31 @@
-# 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
 """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)