Store full preview in array metadata

tiledb/ml/models/_array_metadata.py

@@ -0,0 +1,15 @@
+from enum import Enum, unique
+
+
+@unique
+class ModelArrayMetadata(Enum):
+    """
+    Enum Class that contains all model array metadata.
+    """
+
+    TILEDB_ML_MODEL_ML_FRAMEWORK = "TILEDB_ML_MODEL_ML_FRAMEWORK"
+    TILEDB_ML_MODEL_ML_FRAMEWORK_VERSION = "TILEDB_ML_MODEL_ML_FRAMEWORK_VERSION"
+    TILEDB_ML_MODEL_STAGE = "TILEDB_ML_MODEL_STAGE"
+    TILEDB_ML_MODEL_PYTHON_VERSION = "TILEDB_ML_MODEL_PYTHON_VERSION"
+    TILEDB_ML_MODEL_PREVIEW = "TILEDB_ML_MODEL_PREVIEW"
+    TILEDB_ML_MODEL_VERSION = "TILEDB_ML_MODEL_VERSION"

tiledb/ml/models/_base.py

@@ -25,6 +25,7 @@
 from .. import __version__
 from ._cloud_utils import get_cloud_uri, update_file_properties
 from ._file_properties import ModelFileProperties
+from ._array_metadata import ModelArrayMetadata
 
 Artifact = TypeVar("Artifact")
 Meta = Mapping[str, Any]
@@ -74,6 +75,11 @@ def __init__(
             ModelFileProperties.TILEDB_ML_MODEL_PREVIEW.value: self.preview_short(),
             ModelFileProperties.TILEDB_ML_MODEL_VERSION.value: __version__,
         }
+        # Full/long versions of all properties
+        self._array_metadata = self._file_properties
+        self._array_metadata[
+            ModelArrayMetadata.TILEDB_ML_MODEL_PREVIEW.value
+        ] = self.preview()
 
     @abstractmethod
     def save(self, *, meta: Optional[Meta] = None) -> None:
@@ -166,7 +172,7 @@ def _write_array(
 
         if meta is None:
             meta = {}
-        if not meta.keys().isdisjoint(self._file_properties.keys()):
+        if not meta.keys().isdisjoint(self._array_metadata.keys()):
             raise ValueError(
                 "Please avoid using file property key names as metadata keys!"
             )
@@ -191,7 +197,7 @@ def _write_array(
                 key: np.pad(value, (0, max_len - len(value)))
                 for key, value in one_d_buffers.items()
             }
-            for mapping in meta, self._file_properties:
+            for mapping in meta, self._array_metadata:
                 for key, value in mapping.items():
                     model_array.meta[key] = value