removing the deprecated calls

zenml-io · Nov 13, 2024 · 97ac292 · 97ac292
1 parent a200a13
commit 97ac292
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/template/steps/data_preprocessor.py b/template/steps/data_preprocessor.py
@@ -7,7 +7,7 @@
 from sklearn.preprocessing import MinMaxScaler
 from typing_extensions import Annotated
 from utils.preprocess import ColumnsDropper, DataFrameCaster, NADropper
-from zenml import log_artifact_metadata, step
+from zenml import log_metadata, step
 
 
 @step
@@ -67,7 +67,7 @@ def data_preprocessor(
     dataset_tst = preprocess_pipeline.transform(dataset_tst)
 
     # Log metadata so we can load it in the inference pipeline
-    log_artifact_metadata(
+    log_metadata(
         artifact_name="preprocess_pipeline",
         metadata={"random_state": random_state, "target": target},
     )

diff --git a/template/steps/model_evaluator.py b/template/steps/model_evaluator.py
@@ -4,7 +4,7 @@
 
 import pandas as pd
 from sklearn.base import ClassifierMixin
-from zenml import log_artifact_metadata, step
+from zenml import log_metadata, step
 from zenml.logger import get_logger
 
 logger = get_logger(__name__)
@@ -79,7 +79,7 @@ def model_evaluator(
         for message in messages:
             logger.warning(message)
 
-    log_artifact_metadata(
+    log_metadata(
         metadata={"train_accuracy": float(trn_acc), "test_accuracy": float(tst_acc)},
         artifact_name="sklearn_classifier",
     )