Merge pull request #28 from zenml-io/2024.11.28

Removing the deprecated `log_xxx_metadata` calls
zenml-io · Dec 2, 2024 · 4f9d89b · 4f9d89b
2 parents dadf9be + 1ae14e3
commit 4f9d89b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -56,5 +56,5 @@ jobs:
         with:
           stack-name: ${{ matrix.stack-name }}
           python-version: ${{ matrix.python-version }}
-          ref-zenml: ${{ inputs.ref-zenml || 'develop' }}
+          ref-zenml: ${{ inputs.ref-zenml || 'feature/followup-run-metadata' }}
           ref-template: ${{ inputs.ref-template || github.ref }}
diff --git a/template/steps/training/model_trainer.py b/template/steps/training/model_trainer.py
@@ -13,7 +13,7 @@
     TrainingArguments,
     AutoModelForSequenceClassification,
 )
-from zenml import ArtifactConfig, log_artifact_metadata, step
+from zenml import ArtifactConfig, log_metadata, step
 from zenml.client import Client
 from zenml.integrations.mlflow.experiment_trackers import MLFlowExperimentTracker
 from zenml.logger import get_logger
@@ -135,9 +135,10 @@ def model_trainer(
     eval_results = trainer.evaluate(metric_key_prefix="")
 
     # Log the evaluation results in model control plane
-    log_artifact_metadata(
+    log_metadata(
         metadata={"metrics": eval_results},
         artifact_name="model",
+        infer_artifact=True,
     )
     ### YOUR CODE ENDS HERE ###