Store visualizations for pydantic materializer (#4014)

Author: schustmi

src/zenml/materializers/pydantic_materializer.py

@@ -20,7 +20,7 @@
 
 from pydantic import BaseModel
 
-from zenml.enums import ArtifactType
+from zenml.enums import ArtifactType, VisualizationType
 from zenml.materializers.base_materializer import BaseMaterializer
 from zenml.utils import yaml_utils
 
@@ -85,4 +85,14 @@ def compute_content_hash(self, data: BaseModel) -> Optional[str]:
         hash_.update(json.dumps(json_data, sort_keys=True).encode())
         return hash_.hexdigest()
 
-        return None
+    def save_visualizations(self, data: Any) -> Dict[str, "VisualizationType"]:
+        """Save visualizations for the given data.
+
+        Args:
+            data: The data to save visualizations for.
+
+        Returns:
+            A dictionary of visualization URIs and their types.
+        """
+        data_path = os.path.join(self.uri, DEFAULT_FILENAME)
+        return {data_path.replace("\\", "/"): VisualizationType.JSON}