From 5c3a6f8e154568433a208e39803222f1fe0407c1 Mon Sep 17 00:00:00 2001 From: Hamza Tahir Date: Tue, 14 Feb 2023 08:12:43 +0100 Subject: [PATCH 1/3] Changed artifact type --- .../model_metadata.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/starter/template/{% if use_custom_artifacts %}materializers{% endif %}/model_metadata.py b/starter/template/{% if use_custom_artifacts %}materializers{% endif %}/model_metadata.py index 80a3e7f..5c147ea 100644 --- a/starter/template/{% if use_custom_artifacts %}materializers{% endif %}/model_metadata.py +++ b/starter/template/{% if use_custom_artifacts %}materializers{% endif %}/model_metadata.py @@ -43,7 +43,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. From 439f375b42ce393cfaf6b0a673b0095d9dc9b369 Mon Sep 17 00:00:00 2001 From: Hamza Tahir Date: Tue, 14 Feb 2023 12:16:39 +0500 Subject: [PATCH 2/3] Renamed file --- .../__init__.py | 2 +- .../{model_metadata.py => model_metadata_materializer.py} | 0 2 files changed, 1 insertion(+), 1 deletion(-) rename starter/template/{% if use_custom_artifacts %}materializers{% endif %}/{model_metadata.py => model_metadata_materializer.py} (100%) diff --git a/starter/template/{% if use_custom_artifacts %}materializers{% endif %}/__init__.py b/starter/template/{% if use_custom_artifacts %}materializers{% endif %}/__init__.py index 001a534..07569a9 100644 --- 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/starter/template/{% if use_custom_artifacts %}materializers{% endif %}/model_metadata.py b/starter/template/{% if use_custom_artifacts %}materializers{% endif %}/model_metadata_materializer.py similarity index 100% rename from starter/template/{% if use_custom_artifacts %}materializers{% endif %}/model_metadata.py rename to starter/template/{% if use_custom_artifacts %}materializers{% endif %}/model_metadata_materializer.py From 9d536ca4f832c98cf312fd4fec24d026d028d5c0 Mon Sep 17 00:00:00 2001 From: Hamza Tahir Date: Tue, 14 Feb 2023 12:18:04 +0500 Subject: [PATCH 3/3] Space added --- .../model_metadata_materializer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/starter/template/{% if use_custom_artifacts %}materializers{% endif %}/model_metadata_materializer.py b/starter/template/{% if use_custom_artifacts %}materializers{% endif %}/model_metadata_materializer.py index 5c147ea..320c950 100644 --- a/starter/template/{% if use_custom_artifacts %}materializers{% endif %}/model_metadata_materializer.py +++ b/starter/template/{% if use_custom_artifacts %}materializers{% 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.