1 files changed, 37 insertions, 0 deletions
diff --git a/src/mlia/core/metadata.py b/src/mlia/core/metadata.py
new file mode 100644
index 0000000..f0a0e03
--- /dev/null
+++ b/src/mlia/core/metadata.py
@@ -0,0 +1,37 @@
+# SPDX-FileCopyrightText: Copyright 2023, Arm Limited and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+"""Classes for metadata."""
+from pathlib import Path
+
+from mlia.utils.misc import get_file_checksum
+from mlia.utils.misc import get_pkg_version
+
+
+class Metadata:  # pylint: disable=too-few-public-methods
+    """Base class metadata."""
+
+    def __init__(self, data_name: str) -> None:
+        """Init Metadata."""
+        self.version = self.get_version(data_name)
+
+    def get_version(self, data_name: str) -> str:
+        """Get version of the python package."""
+        return get_pkg_version(data_name)
+
+
+class MLIAMetadata(Metadata):  # pylint: disable=too-few-public-methods
+    """MLIA metadata."""
+
+
+class ModelMetadata:  # pylint: disable=too-few-public-methods
+    """Model metadata."""
+
+    def __init__(self, path_name: Path) -> None:
+        """Init ModelMetadata."""
+        self.model_name = path_name.name
+        self.path_name = path_name
+        self.checksum = self.get_checksum()
+
+    def get_checksum(self) -> str:
+        """Get checksum of the model."""
+        return get_file_checksum(self.path_name)