Merge pull request #2 from zenml-io/htahir1-patch-1

Changed artifact type
zenml-io · Feb 14, 2023 · 83a289e · 83a289e
2 parents 88e5d96 + 9d536ca
commit 83a289e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/starter/template/{% if use_custom_artifacts %}materializers{% endif %}/__init__.py b/starter/template/{% if use_custom_artifacts %}materializers{% endif %}/__init__.py
@@ -1,6 +1,6 @@
 {% include 'templates/license_header' %}
 
-from materializers.model_metadata import ModelMetadataMaterializer
+from materializers.model_metadata_materializer import ModelMetadataMaterializer
 
 __all__ = [
     "ModelMetadataMaterializer",

diff --git a/...aterializers{% endif %}/model_metadata.py → ...% endif %}/model_metadata_materializer.py b/...aterializers{% endif %}/model_metadata.py → ...% endif %}/model_metadata_materializer.py
@@ -10,6 +10,7 @@
 from zenml.io import fileio
 from zenml.materializers.base_materializer import BaseMaterializer
 
+
 class ModelMetadataMaterializer(BaseMaterializer):
     """Custom materializer for the `ModelMetadata` artifact data type.
 
@@ -43,7 +44,7 @@ class ModelMetadataMaterializer(BaseMaterializer):
     # This needs to point to the artifact data type(s) associated with the
     # materializer 
     ASSOCIATED_TYPES = (ModelMetadata,)
-    ASSOCIATED_ARTIFACT_TYPE = ArtifactType.DATA
+    ASSOCIATED_ARTIFACT_TYPE = ArtifactType.STATISTICS
 
     def save(self, model_metadata: ModelMetadata) -> None:
         """Save (de-materialize) a model metadata artifact to the artifact store.